File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -15,10 +15,10 @@ elif [ "${build_arch}" = "aarch64" ]; then
15
15
[[ " ${cuda_version} " == 12.8.* || " ${cuda_version} " == 12.9.* ]] && build_capability=" 75;80;90;100;120"
16
16
else
17
17
# By default, target Maxwell through Hopper.
18
- build_capability=" 50;52;60;61 ;70;75;80;86;89;90"
18
+ build_capability=" 50;60 ;70;75;80;86;89;90"
19
19
20
- # CUDA 12.8+: Add sm100 and sm120; remove < sm75 to align with PyTorch 2.7 +cu128 minimum
21
- [[ " ${cuda_version} " == 12.8.* || " ${cuda_version} " == 12.9.* ]] && build_capability=" 75;80;86;89;90;100;120"
20
+ # CUDA 12.8+: Add sm100 and sm120; remove < sm70 to align with PyTorch 2.8 +cu128 minimum
21
+ [[ " ${cuda_version} " == 12.8.* || " ${cuda_version} " == 12.9.* ]] && build_capability=" 70; 75;80;86;89;90;100;120"
22
22
fi
23
23
24
24
[[ " ${build_os} " = windows-* ]] && python3 -m pip install ninja
You can’t perform that action at this time.
0 commit comments