From e88396e5edf2bb50776985b4db0e358e95ee1df2 Mon Sep 17 00:00:00 2001
From: Ken Raffenetti <raffenet@users.noreply.github.com>
Date: Thu, 5 May 2022 10:12:21 -0500
Subject: yaksa: Add support for cuda_arch variants (#30471)
---
var/spack/repos/builtin/packages/yaksa/package.py | 5 +++++
1 file changed, 5 insertions(+)
(limited to 'var')
diff --git a/var/spack/repos/builtin/packages/yaksa/package.py b/var/spack/repos/builtin/packages/yaksa/package.py
index b8ded1866c..25f872f45e 100644
--- a/var/spack/repos/builtin/packages/yaksa/package.py
+++ b/var/spack/repos/builtin/packages/yaksa/package.py
@@ -41,7 +41,12 @@ class Yaksa(AutotoolsPackage, CudaPackage, ROCmPackage):
def configure_args(self):
spec = self.spec
config_args = []
+
config_args += self.with_or_without('cuda', activation_value='prefix')
+ if '+cuda' in spec:
+ cuda_archs = spec.variants['cuda_arch'].value
+ if 'none' not in cuda_archs:
+ config_args.append('--with-cuda-sm={0}'.format(",".join(cuda_archs)))
if '+rocm' in spec:
config_args.append('--with-hip={0}'.format(spec['hip'].prefix))
--
cgit v1.2.3-70-g09d2