spfft: pass cuda_arch to cmake (#30295)

2022-04-26 14:59:48 +02:00 · 2022-04-26 14:59:48 +02:00 · 3e51304b68
commit 3e51304b68
parent 62abbeaf6f
1 changed files with 4 additions and 0 deletions
--- a/var/spack/repos/builtin/packages/spfft/package.py
+++ b/var/spack/repos/builtin/packages/spfft/package.py
@ -86,6 +86,10 @@ def cmake_args(self):
        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 += [