Skip to content

Commit d797946

Browse files
Do not unnecessarily install CUDA for ROCm (#3837)
Summary: ROCm does not require CUDA, this change stops installing it. This is also likely to fix #3787 which is coming in a separate change. Differential Revision: D62283602
1 parent e261725 commit d797946

1 file changed

Lines changed: 7 additions & 6 deletions

File tree

.github/actions/build_cmake/action.yml

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -51,13 +51,14 @@ runs:
5151
conda install -y -q mkl=2023 mkl-devel=2023
5252
fi
5353
54-
# install CUDA packages
55-
if [ "${{ inputs.gpu }}" = "ON" ] && [ "${{ inputs.raft }}" = "OFF" ]; then
54+
# no CUDA needed for ROCm so skip this
55+
if [ "${{ inputs.rocm }}" = "ON" ]; then
56+
:
57+
# regular CUDA for GPU builds
58+
elif [ "${{ inputs.gpu }}" = "ON" ] && [ "${{ inputs.raft }}" = "OFF" ]; then
5659
conda install -y -q cuda-toolkit -c "nvidia/label/cuda-12.4.0"
57-
fi
58-
59-
# install RAFT packages
60-
if [ "${{ inputs.raft }}" = "ON" ]; then
60+
# and CUDA from RAFT channel for RAFT builds
61+
elif [ "${{ inputs.raft }}" = "ON" ]; then
6162
conda install -y -q libraft cuda-version=12.4 cuda-toolkit -c rapidsai-nightly -c "nvidia/label/cuda-12.4.0" -c conda-forge
6263
fi
6364

0 commit comments

Comments
 (0)