From 3e51304b68724a89c5048a293d926a55925b8aef Mon Sep 17 00:00:00 2001
From: Simon Pintarelli <1237199+simonpintarelli@users.noreply.github.com>
Date: Tue, 26 Apr 2022 14:59:48 +0200
Subject: [PATCH] spfft: pass cuda_arch to cmake (#30295)

---
 var/spack/repos/builtin/packages/spfft/package.py | 4 ++++
 1 file changed, 4 insertions(+)

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