Skip to content

Commit 0785b66

Browse files
committed
Add support for CUDA 12 and Hopper.
1 parent aa17817 commit 0785b66

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/configure

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,6 +283,7 @@ Either your CUDA is too new or too old."
283283
GCC_VER=$($CXX -dumpversion)
284284
GCC_VER_NUM=$(echo $GCC_VER | sed 's/\./ /g' | xargs printf "%d%02d%02d")
285285
case $CUDA_VERSION in
286+
# Update this list by consulting https://gist.github.com/ax3l/9489132
286287
# Disabling CUDA 7 and CUDA 8 because we now use C++14 to compile CUDA
287288
# code. It is still possible to use those cuda versions by switching
288289
# back to C++11 in src/makefiles/cuda_64bit.mk and use CUB <= 1.8.0.
@@ -317,7 +318,13 @@ Either your CUDA is too new or too old."
317318
11_*)
318319
MIN_UNSUPPORTED_GCC_VER="12.0"
319320
MIN_UNSUPPORTED_GCC_VER_NUM=120000
320-
;;
321+
CUSOLVER=true
322+
;;
323+
12_*)
324+
MIN_UNSUPPORTED_GCC_VER="12.2"
325+
MIN_UNSUPPORTED_GCC_VER_NUM=122000
326+
CUSOLVER=true
327+
;;
321328
*)
322329
failure "Unsupported CUDA version ${CUDA_VERSION}.
323330
Please open an issue at https://github.com/kaldi-asr/kaldi/issues and include\
@@ -345,6 +352,7 @@ Please open an issue at https://github.com/kaldi-asr/kaldi/issues and include\
345352
10_*) CUDA_ARCH="-gencode arch=compute_30,code=sm_30 -gencode arch=compute_35,code=sm_35 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_52,code=sm_52 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75" ;;
346353
11_0) CUDA_ARCH="-gencode arch=compute_35,code=sm_35 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_52,code=sm_52 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80" ;;
347354
11_*) CUDA_ARCH="-gencode arch=compute_35,code=sm_35 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_52,code=sm_52 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86" ;;
355+
12_*) CUDA_ARCH="-gencode arch=compute_50,code=sm_50 -gencode arch=compute_52,code=sm_52 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_90,code=sm_90" ;;
348356
*) failure \
349357
"Unsupported CUDA version ${CUDA_VERSION}. Please open an" \
350358
"issue at https://github.com/kaldi-asr/kaldi/issues and" \

0 commit comments

Comments
 (0)