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
1515 [[ " ${cuda_version} " == 12.8.* || " ${cuda_version} " == 12.9.* ]] && build_capability=" 75;80;90;100;120"
1616else
1717 # 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"
1919
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"
2222fi
2323
2424[[ " ${build_os} " = windows-* ]] && python3 -m pip install ninja
You can’t perform that action at this time.
0 commit comments