Merge pull request #81 from GRTLCollaboration/bugfix/80_missing_gpu_deps

CI: Add missing CUDA and ROCm dependencies
GRTLCollaboration · Jan 7, 2025 · 03e7a2e · 03e7a2e
2 parents 9361fcf + 92409b3
commit 03e7a2e
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/.github/workflows/gpu-build.yml b/.github/workflows/gpu-build.yml
@@ -90,21 +90,24 @@ 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
           roctracer-dev
           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 }}