Skip to content

Commit

Permalink
CI: Add missing CUDA and ROCm dependencies
Browse files Browse the repository at this point in the history
This fixes #80.

Also refactor the list of CUDA packages in a very minor way to reduce
duplication.
  • Loading branch information
mirenradia committed Dec 23, 2024
1 parent 268833d commit 3c689ef
Showing 1 changed file with 10 additions and 7 deletions.
17 changes: 10 additions & 7 deletions .github/workflows/gpu-build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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 }}
Expand Down

0 comments on commit 3c689ef

Please sign in to comment.