Skip to content

Commit

Permalink
add cuda12 option for aarch64 (#4949)
Browse files Browse the repository at this point in the history
  • Loading branch information
jtrmal authored Oct 4, 2024
1 parent d9ab046 commit 4a8b7f6
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 2 deletions.
2 changes: 0 additions & 2 deletions .github/workflows/c-cpp.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,6 @@ jobs:
- uses: actions/checkout@v4
- name: Install sox
run: sudo apt-get install -y sox intel-mkl
- name: Install python2
run: sudo apt-get install -y python2
- name: ccache
uses: hendrikmuhs/[email protected]
with:
Expand Down
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 4a8b7f6

Please sign in to comment.