diff options
| author | Sreenivasa Murthy Kolam <67086238+srekolam@users.noreply.github.com> | 2021-04-10 05:17:21 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-04-10 11:17:21 +0200 |
| commit | 291a2b541b8d82945fcf7edf5083ecc88d1f322d (patch) | |
| tree | 91de8698d55a7912e8ecceb9990f23bd2edafc9b | |
| parent | 391ab10337249b9f5bfe8817a8a450f2e605d607 (diff) | |
| download | spack-291a2b541b8d82945fcf7edf5083ecc88d1f322d.tar.gz spack-291a2b541b8d82945fcf7edf5083ecc88d1f322d.tar.bz2 spack-291a2b541b8d82945fcf7edf5083ecc88d1f322d.tar.xz spack-291a2b541b8d82945fcf7edf5083ecc88d1f322d.zip | |
fix 22802- fix incorrect arch list for rocblas (#22881)
| -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): |