From 3c689ef94f81f2b166391fc47b0c5f8a26bc9fd1 Mon Sep 17 00:00:00 2001
From: Miren Radia <miren_radia@yahoo.co.uk>
Date: Fri, 20 Dec 2024 15:49:02 +0000
Subject: [PATCH] CI: Add missing CUDA and ROCm dependencies

This fixes #80.

Also refactor the list of CUDA packages in a very minor way to reduce
duplication.
---
 .github/workflows/gpu-build.yml | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/gpu-build.yml b/.github/workflows/gpu-build.yml
index 98d2bfe1..14e5f8c8 100644
--- a/.github/workflows/gpu-build.yml
+++ b/.github/workflows/gpu-build.yml
@@ -90,14 +90,16 @@ jobs:
       # These have been copied/adapted from AMReX's CI dependencies
       run: |
         if [[ "${{ matrix.gpu-backend }}" == "CUDA" ]]; then
+          CUDA_DASHED_VERSION="${{ env.CUDA_MAJOR_VERSION }}-${{ env.CUDA_MINOR_VERSION }}"
           PACKAGES="libopenmpi-dev
-          cuda-command-line-tools-${{ env.CUDA_MAJOR_VERSION }}-${{ env.CUDA_MINOR_VERSION }}
-          cuda-compiler-${{ env.CUDA_MAJOR_VERSION }}-${{ env.CUDA_MINOR_VERSION }}
-          cuda-cupti-dev-${{ env.CUDA_MAJOR_VERSION }}-${{ env.CUDA_MINOR_VERSION }}
-          cuda-minimal-build-${{ env.CUDA_MAJOR_VERSION }}-${{ env.CUDA_MINOR_VERSION }}
-          cuda-nvml-dev-${{ env.CUDA_MAJOR_VERSION }}-${{ env.CUDA_MINOR_VERSION }}
-          cuda-nvtx-${{ env.CUDA_MAJOR_VERSION }}-${{ env.CUDA_MINOR_VERSION }}
-          libcurand-dev-${{ env.CUDA_MAJOR_VERSION }}-${{ env.CUDA_MINOR_VERSION }}"
+          cuda-command-line-tools-${CUDA_DASHED_VERSION}
+          cuda-compiler-${CUDA_DASHED_VERSION}
+          cuda-cupti-dev-${CUDA_DASHED_VERSION}
+          cuda-minimal-build-${CUDA_DASHED_VERSION}
+          cuda-nvml-dev-${CUDA_DASHED_VERSION}
+          cuda-nvtx-${CUDA_DASHED_VERSION}
+          libcurand-dev-${CUDA_DASHED_VERSION}
+          libcusparse-dev-${CUDA_DASHED_VERSION}"
         elif [[ "${{ matrix.gpu-backend }}" == "HIP" ]]; then
           PACKAGES="libopenmpi-dev
           rocm-dev
@@ -105,6 +107,7 @@ jobs:
           rocprofiler-dev
           rocrand-dev
           rocprim-dev
+          rocsparse-dev
           hiprand-dev"
         elif [[ "${{ matrix.gpu-backend }}" == "SYCL" ]]; then
           PACKAGES="intel-oneapi-compiler-dpcpp-cpp-${{ env.ONEAPI_COMP_VERSION }}