Skip to content

Commit

Permalink
add cuda12 option for aarch64
Browse files Browse the repository at this point in the history
  • Loading branch information
jtrmal committed Oct 2, 2024
1 parent d9ab046 commit c06f32e
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/configure
Original file line number Diff line number Diff line change
Expand Up @@ -434,6 +434,7 @@ Please open an issue at https://github.com/kaldi-asr/kaldi/issues and include\
#8_*) CUDA_ARCH="-gencode arch=compute_53,code=sm_53 -gencode arch=compute_62,code=sm_62" ;;
9_*) CUDA_ARCH="-gencode arch=compute_53,code=sm_53 -gencode arch=compute_62,code=sm_62" ;;
10_*|11_*) CUDA_ARCH="-gencode arch=compute_53,code=sm_53 -gencode arch=compute_62,code=sm_62 -gencode arch=compute_72,code=sm_72" ;;
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" ;;
*) echo "Unsupported CUDA_VERSION (CUDA_VERSION=$CUDA_VERSION), please report it to Kaldi mailing list, together with 'nvcc -h' or 'ptxas -h' which lists allowed -gencode values..."; exit 1 ;;
esac
;;
Expand Down

0 comments on commit c06f32e

Please sign in to comment.