diff --git a/var/spack/repos/builtin/packages/spfft/package.py b/var/spack/repos/builtin/packages/spfft/package.py index 59cf2422f5a..00381428ed2 100644 --- 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 += [