diff options
author | Simon Pintarelli <1237199+simonpintarelli@users.noreply.github.com> | 2022-04-26 14:59:48 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-04-26 14:59:48 +0200 |
commit | 3e51304b68724a89c5048a293d926a55925b8aef (patch) | |
tree | ac5292894dd6e20599be118e9843d23cb4848e0c | |
parent | 62abbeaf6f9e6aaa8f34baa2ef1ee57139577ea8 (diff) | |
download | spack-3e51304b68724a89c5048a293d926a55925b8aef.tar.gz spack-3e51304b68724a89c5048a293d926a55925b8aef.tar.bz2 spack-3e51304b68724a89c5048a293d926a55925b8aef.tar.xz spack-3e51304b68724a89c5048a293d926a55925b8aef.zip |
spfft: pass cuda_arch to cmake (#30295)
-rw-r--r-- | var/spack/repos/builtin/packages/spfft/package.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/spfft/package.py b/var/spack/repos/builtin/packages/spfft/package.py index 59cf2422f5..00381428ed 100644 --- a/var/spack/repos/builtin/packages/spfft/package.py +++ b/var/spack/repos/builtin/packages/spfft/package.py @@ -86,6 +86,10 @@ class Spfft(CMakePackage, CudaPackage): if spec.satisfies('+cuda'): args += ["-DSPFFT_GPU_BACKEND=CUDA"] + cuda_arch = self.spec.variants['cuda_arch'].value + if cuda_arch[0] != 'none': + args += [self.define('CMAKE_CUDA_ARCHITECTURES', cuda_arch)] + if spec.satisfies('+rocm'): archs = ",".join(self.spec.variants['amdgpu_target'].value) args += [ |