Skip to content

Commit 805825f

Browse files
committed
Add CUDA 11.8 branch for sm_90
1 parent 05e4465 commit 805825f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

recipe/build-lib.sh

+6-1
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,12 @@ if [ ${cuda_compiler_version} != "None" ]; then
1212
# docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#gpu-feature-list
1313

1414
ARCHES=(53 62 72)
15-
if [ $(version2int $cuda_compiler_version) -ge $(version2int "11.1") ]; then
15+
if [ $(version2int $cuda_compiler_version) -ge $(version2int "11.8") ]; then
16+
# Hopper support for H100 (sm_90) needs cuda >= 11.8
17+
LATEST_ARCH=90
18+
# ARCHES does not contain LATEST_ARCH; see usage below
19+
ARCHES=( "${ARCHES[@]}" 75 80 86)
20+
elif [ $(version2int $cuda_compiler_version) -ge $(version2int "11.1") ]; then
1621
# Ampere support for GeForce 30 (sm_86) needs cuda >= 11.1
1722
LATEST_ARCH=86
1823
# ARCHES does not contain LATEST_ARCH; see usage below

0 commit comments

Comments
 (0)