zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

kokkos: add ampere80 cuda arch (#22397)

Browse Source 
Adding support for Ampere (SM_80) GPUs.
This commit is contained in:
Scot Halverson 2021-03-20 05:54:48 -06:00 committed by GitHub
parent c055ffc79b
commit e1abb5cbc8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
var/spack/repos/builtin/packages/kokkos/package.py
View File

@ -140,6 +140,7 @@ class Kokkos(CMakePackage, CudaPackage, ROCmPackage):
"70": 'volta70', "70": 'volta70',
"72": 'volta72', "72": 'volta72',
"75": 'turing75', "75": 'turing75',
"80": 'ampere80',
} }
cuda_arches = spack_cuda_arch_map.values() cuda_arches = spack_cuda_arch_map.values()
conflicts("+cuda", when="cuda_arch=none") conflicts("+cuda", when="cuda_arch=none")