We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 77db267 commit 18c07bdCopy full SHA for 18c07bd
.github/scripts/build-cuda.sh
@@ -6,14 +6,14 @@ declare cuda_targets
6
7
set -xeuo pipefail
8
9
-if [ -n "${cuda_targets}" ]; then
+if [[ -v cuda_targets ]; then
10
+ build_capability="${cuda_targets}"
11
+else
12
# By default, target Maxwell through Hopper.
13
build_capability="50;52;60;61;70;75;80;86;89;90"
14
15
# CUDA 12.8: Add sm100 and sm120; remove < sm75 to align with PyTorch 2.7+cu128 minimum
16
[[ "${cuda_version}" == 12.8.* ]] && build_capability="75;80;86;89;90;100;120"
-else
- build_capability="${cuda_targets}"
17
fi
18
19
[[ "${build_os}" = windows-* ]] && python3 -m pip install ninja
0 commit comments