diff options
| -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 += [ |