diff options
author | Ken Raffenetti <raffenet@users.noreply.github.com> | 2022-05-05 10:12:21 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-05-05 08:12:21 -0700 |
commit | e88396e5edf2bb50776985b4db0e358e95ee1df2 (patch) | |
tree | 1d4e8fd015f8e10c94efd4b9a8025b156090f658 /var | |
parent | dcd2f8a4ed191a40dbc6e2edefc366ec47b9673e (diff) | |
download | spack-e88396e5edf2bb50776985b4db0e358e95ee1df2.tar.gz spack-e88396e5edf2bb50776985b4db0e358e95ee1df2.tar.bz2 spack-e88396e5edf2bb50776985b4db0e358e95ee1df2.tar.xz spack-e88396e5edf2bb50776985b4db0e358e95ee1df2.zip |
yaksa: Add support for cuda_arch variants (#30471)
Diffstat (limited to 'var')
-rw-r--r-- | var/spack/repos/builtin/packages/yaksa/package.py | 5 |
1 files changed, 5 insertions, 0 deletions
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)) |