Commit 805825f 1 parent 05e4465 commit 805825f Copy full SHA for 805825f
File tree 1 file changed +6
-1
lines changed
1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -12,7 +12,12 @@ if [ ${cuda_compiler_version} != "None" ]; then
12
12
# docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#gpu-feature-list
13
13
14
14
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
16
21
# Ampere support for GeForce 30 (sm_86) needs cuda >= 11.1
17
22
LATEST_ARCH=86
18
23
# ARCHES does not contain LATEST_ARCH; see usage below
You can’t perform that action at this time.
0 commit comments