diff options
| -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)) |