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

petsc: pass in 'cuda_arch' to configure via --with-cuda-gencodearch [or CUDAFLAGS for older releases] (#24962)

Browse Source
This commit is contained in:
Satish Balay 2021-07-20 02:48:48 -05:00 committed by GitHub
parent 330507f329
commit 2c87992506
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
var/spack/repos/builtin/packages/petsc/package.py
View File

@ -431,6 +431,15 @@ def install(self, spec, prefix):
library=petsclibname, path=spec[library].prefix)
)
if '+cuda' in spec:
if not spec.satisfies('cuda_arch=none'):
cuda_arch = spec.variants['cuda_arch'].value
if spec.satisfies('@3.14:'):
options.append('--with-cuda-gencodearch={0}'.format(cuda_arch[0]))
else:
options.append('CUDAFLAGS=-gencode arch=compute_{0},code=sm_{0}'
.format(cuda_arch[0]))
# PETSc does not pick up SuperluDist from the dir as they look for
# superlu_dist_4.1.a
if 'superlu-dist' in spec: