Skip to content

Commit 28b585b

Browse files
authored
more granularity
1 parent b22711c commit 28b585b

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

.github/actions/fetch_ctk/action.yml

+6-2
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ inputs:
1717
description: "A list of the CTK components to install as a comma-separated list. e.g. 'cuda_nvcc,cuda_nvrtc,cuda_cudart'"
1818
required: false
1919
type: string
20-
default: "cuda_nvcc,cuda_cudart,cuda_nvrtc,cuda_profiler_api,cuda_cccl,cuda_sanitizer_api,libnvjitlink,libcublas"
20+
default: "cuda_nvcc,cuda_cudart,cuda_nvrtc,cuda_profiler_api,cuda_cccl,cuda_sanitizer_api,libnvjitlink"
2121

2222
runs:
2323
using: composite
@@ -102,7 +102,11 @@ runs:
102102
if [[ "$item" == "cuda_sanitizer_api" ]]; then
103103
# Always use latest CTK for cuda_sanitizer_api
104104
# FIXME: Automatically track latest CTK version
105-
ctk_version="12.8.1"
105+
CUDA_MAJOR="$(cut -d '.' -f 1 <<< ${{ inputs.cuda-version }})"
106+
if [[ "$CUDA_MAJOR" == "12" ]]; then
107+
# TODO: Automatically track latest CTK minor version
108+
ctk_version="12.8.0"
109+
fi
106110
fi
107111
populate_cuda_path "$item" "$ctk_version"
108112
done

0 commit comments

Comments
 (0)