Skip to content

GitLab

Commit ec192295 authored by Matthew Douglas's avatar Matthew Douglas
Browse files

Add Volta support in cu128/cu129 builds

parent 33449ee4
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
.github/scripts/build-cuda.sh
View file @ ec192295
...@@ -15,10 +15,10 @@ elif [ "${build_arch}" = "aarch64" ]; then ...@@ -15,10 +15,10 @@ elif [ "${build_arch}" = "aarch64" ]; then
[[ "${cuda_version}" == 12.8.* || "${cuda_version}" == 12.9.* ]] && build_capability="75;80;90;100;120" [[ "${cuda_version}" == 12.8.* || "${cuda_version}" == 12.9.* ]] && build_capability="75;80;90;100;120"
else else
# By default, target Maxwell through Hopper. # By default, target Maxwell through Hopper.
build_capability="50;52;60;61;70;75;80;86;89;90" build_capability="50;60;70;75;80;86;89;90"
# CUDA 12.8+: Add sm100 and sm120; remove < sm75 to align with PyTorch 2.7+cu128 minimum # CUDA 12.8+: Add sm100 and sm120; remove < sm70 to align with PyTorch 2.8+cu128 minimum
[[ "${cuda_version}" == 12.8.* || "${cuda_version}" == 12.9.* ]] && build_capability="75;80;86;89;90;100;120" [[ "${cuda_version}" == 12.8.* || "${cuda_version}" == 12.9.* ]] && build_capability="70;75;80;86;89;90;100;120"
fi fi
[[ "${build_os}" = windows-* ]] && python3 -m pip install ninja [[ "${build_os}" = windows-* ]] && python3 -m pip install ninja
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment