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
[[ "${cuda_version}" == 12.8.* || "${cuda_version}" == 12.9.* ]] && build_capability="75;80;90;100;120"
else
# 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_version}" == 12.8.* || "${cuda_version}" == 12.9.* ]] && build_capability="75;80;86;89;90;100;120"
# 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="70;75;80;86;89;90;100;120"
fi
[[ "${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