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

fix 22802- fix incorrect arch list for rocblas (#22881)

Browse Source
This commit is contained in:
Sreenivasa Murthy Kolam 2021-04-10 05:17:21 -04:00 committed by GitHub
parent 391ab10337
commit 291a2b541b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
var/spack/repos/builtin/packages/migraphx/package.py
View File

@ -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']

6
var/spack/repos/builtin/packages/rocblas/package.py
View File

@ -109,7 +109,6 @@ def cmake_args(self):
'-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 @@ def cmake_args(self):
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

6
var/spack/repos/builtin/packages/rocm-tensile/package.py
View File

@ -62,7 +62,6 @@ def cmake_args(self):
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 @@ def cmake_args(self):
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):