Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "44460c9c0b2948c5d5d5b10053c31c13c629bb4a"
Unverified Commit e54dc125 authored by Matthew Douglas's avatar Matthew Douglas Committed by GitHub
Browse files

Merge pull request #1715 from bitsandbytes-foundation/adjust-cuda-build 

Add Volta support in cu128/cu129 builds
parents 33449ee4 ec192295
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
.github/scripts/build-cuda.sh
View file @ e54dc125
...@@ -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