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 /var | |
| 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)
Diffstat (limited to 'var')
| -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 += [ |