diff options
-rw-r--r-- | var/spack/repos/builtin/packages/migraphx/package.py | 1 | ||||
-rw-r--r-- | var/spack/repos/builtin/packages/rocblas/package.py | 6 | ||||
-rw-r--r-- | var/spack/repos/builtin/packages/rocm-tensile/package.py | 6 |
3 files changed, 11 insertions, 2 deletions
diff --git a/var/spack/repos/builtin/packages/migraphx/package.py b/var/spack/repos/builtin/packages/migraphx/package.py index 47ac2c2b70..3b17d1b86f 100644 --- a/var/spack/repos/builtin/packages/migraphx/package.py +++ b/var/spack/repos/builtin/packages/migraphx/package.py @@ -10,6 +10,7 @@ class Migraphx(CMakePackage): """ AMD's graph optimization engine.""" homepage = "https://github.com/ROCmSoftwarePlatform/AMDMIGraphX" + git = "https://github.com/ROCmSoftwarePlatform/AMDMIGraphX.git" url = "https://github.com/ROCmSoftwarePlatform/AMDMIGraphX/archive/rocm-4.0.0.tar.gz" maintainers = ['srekolam', 'arjun-raj-kuppala'] diff --git a/var/spack/repos/builtin/packages/rocblas/package.py b/var/spack/repos/builtin/packages/rocblas/package.py index ab87cbdeae..2d5f13c263 100644 --- a/var/spack/repos/builtin/packages/rocblas/package.py +++ b/var/spack/repos/builtin/packages/rocblas/package.py @@ -109,7 +109,6 @@ class Rocblas(CMakePackage): '-DBUILD_WITH_TENSILE=ON', '-DTensile_TEST_LOCAL_PATH={0}'.format(tensile), '-DTensile_COMPILER=hipcc', - '-DTensile_ARCHITECTURE={0}'.format(arch), '-DTensile_LOGIC=asm_full', '-DTensile_CODE_OBJECT_VERSION=V3', '-DBUILD_WITH_TENSILE_HOST={0}'.format( @@ -120,4 +119,9 @@ class Rocblas(CMakePackage): if '@3.7.0:' in self.spec: args.append('-DTensile_LIBRARY_FORMAT=msgpack') + if self.spec.satisfies('@4.1.0:'): + if arch == 'gfx906' or arch == 'gfx908': + arch = arch + ':xnack-' + args.append('-DTensile_ARCHITECTURE={0}'.format(arch)) + return args diff --git a/var/spack/repos/builtin/packages/rocm-tensile/package.py b/var/spack/repos/builtin/packages/rocm-tensile/package.py index f9de247f57..8b07e1bbd0 100644 --- a/var/spack/repos/builtin/packages/rocm-tensile/package.py +++ b/var/spack/repos/builtin/packages/rocm-tensile/package.py @@ -62,7 +62,6 @@ class RocmTensile(CMakePackage): args = [ '-Damd_comgr_DIR={0}'.format(self.spec['comgr'].prefix), '-DTensile_COMPILER=hipcc', - '-DTensile_ARCHITECTURE={0}'.format(arch), '-DTensile_LOGIC=asm_full', '-DTensile_CODE_OBJECT_VERSION=V3', '-DBoost_USE_STATIC_LIBS=OFF', @@ -75,6 +74,11 @@ class RocmTensile(CMakePackage): if '@3.7.0:' in self.spec: args.append('-DTensile_LIBRARY_FORMAT=msgpack') + if self.spec.satisfies('@4.1.0:'): + if arch == 'gfx906' or arch == 'gfx908': + arch = arch + ':xnack-' + args.append('-DTensile_ARCHITECTURE={0}'.format(arch)) + return args def install(self, spec, prefix): |