zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

yaksa: add ROCm variant (#30222)

Browse Source
This commit is contained in:
Ken Raffenetti 2022-05-03 09:52:31 -05:00 committed by GitHub
parent a6dcce4cf2
commit 55bbbe8657
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
var/spack/repos/builtin/packages/yaksa/package.py
View File

@ -6,7 +6,7 @@
from spack import *
class Yaksa(AutotoolsPackage, CudaPackage):
class Yaksa(AutotoolsPackage, CudaPackage, ROCmPackage):
"""Yaksa is a high-performance datatype engine for expressing,
managing and manipulating data present in noncontiguous memory
regions. It provides portable abstractions for structured
@ -39,7 +39,16 @@ def autoreconf(self, spec, prefix):
sh('autogen.sh')
def configure_args(self):
spec = self.spec
config_args = []
config_args += self.with_or_without('cuda', activation_value='prefix')
if '+rocm' in spec:
config_args.append('--with-hip={0}'.format(spec['hip'].prefix))
rocm_archs = spec.variants['amdgpu_target'].value
if 'none' not in rocm_archs:
config_args.append('--with-hip-sm={0}'.format(",".join(rocm_archs)))
else:
config_args.append('--without-hip')
return config_args