NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add 89-real to list of architectures built

Browse Source
This commit is contained in:
Nader Al Awar
2026-01-30 13:02:42 -06:00
parent a5ad480dfe
commit 7b5887a4a6
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ci/build_cuda_bench_wheel_for_cuda.sh
View File

@@ -34,13 +34,13 @@ if [[ -z "${CUDAARCHS:-}" ]]; then
}
if version_ge "${cuda_version}" "13.0"; then
CUDAARCHS="75-real;80-real;86-real;90a-real;100f-real;120a-real;120-virtual"
CUDAARCHS="75-real;80-real;86-real;89-real;90a-real;100f-real;120a-real;120-virtual"
elif version_ge "${cuda_version}" "12.9"; then
CUDAARCHS="70-real;75-real;80-real;86-real;90a-real;100f-real;120a-real;120-virtual"
CUDAARCHS="70-real;75-real;80-real;86-real;89-real;90a-real;100f-real;120a-real;120-virtual"
else
CUDAARCHS="70-real;75-real;80-real;86-real;90a-real;90-virtual"
CUDAARCHS="70-real;75-real;80-real;86-real;89-real;90a-real;90-virtual"
if version_ge "${cuda_version}" "12.8"; then
CUDAARCHS="70-real;75-real;80-real;86-real;90a-real;100-real;120a-real;120-virtual"
CUDAARCHS="70-real;75-real;80-real;86-real;89-real;90a-real;100-real;120a-real;120-virtual"
fi
fi
fi