diff --git a/var/spack/repos/builtin/packages/yaksa/package.py b/var/spack/repos/builtin/packages/yaksa/package.py index b8ded1866c3..25f872f45ef 100644 --- a/var/spack/repos/builtin/packages/yaksa/package.py +++ b/var/spack/repos/builtin/packages/yaksa/package.py @@ -41,7 +41,12 @@ def autoreconf(self, spec, prefix): 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))