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

Bump rocblas to 3.8.0 (#19027)

Browse Source 
* Add rocblas 3.8.0 and add all Tensile deps

* Deploy rocm_smi to the bin/ folder so that it is in $PATH

* BUILD_WITH_TENSILE_HOST=ON on 3.7.0+ and fix flake8
This commit is contained in:
Harmen Stoppels 2020-10-02 18:14:12 +02:00 committed by GitHub
parent d1ae087960
commit 0e8be35c25
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,6 +15,7 @@ class Rocblas(CMakePackage):
maintainers = ['haampie'] maintainers = ['haampie']
version('3.8.0', sha256='568a9da0360349b1b134d74cc67cbb69b43c06eeca7c33b50072cd26cd3d8900')
version('3.7.0', sha256='9425db5f8e8b6f7fb172d09e2a360025b63a4e54414607709efc5acb28819642') version('3.7.0', sha256='9425db5f8e8b6f7fb172d09e2a360025b63a4e54414607709efc5acb28819642')
version('3.5.0', sha256='8560fabef7f13e8d67da997de2295399f6ec595edfd77e452978c140d5f936f0') version('3.5.0', sha256='8560fabef7f13e8d67da997de2295399f6ec595edfd77e452978c140d5f936f0')
@ -24,11 +25,14 @@ class Rocblas(CMakePackage):
depends_on('cmake@3:', type='build') depends_on('cmake@3:', type='build')
for ver in ['3.5.0', '3.7.0']: for ver in ['3.5.0', '3.7.0', '3.8.0']:
depends_on('rocm-cmake@' + ver, type='build', when='@' + ver) depends_on('rocm-cmake@' + ver, type='build', when='@' + ver)
depends_on('rocm-device-libs@' + ver, type='build', when='@' + ver) depends_on('rocm-device-libs@' + ver, type='build', when='@' + ver)
depends_on('hip@' + ver, when='@' + ver) depends_on('hip@' + ver, when='@' + ver)
depends_on('comgr@' + ver, type='build', when='@' + ver) depends_on('comgr@' + ver, type='build', when='@' + ver)
# used in Tensile
depends_on('rocm-smi@' + ver, type='build', when='@' + ver)
depends_on('llvm-amdgpu@' + ver, type='build', when='@' + ver)
# This is the default library format since 3.7.0 # This is the default library format since 3.7.0
depends_on('msgpack-c@3:', when='@3.7:') depends_on('msgpack-c@3:', when='@3.7:')
@ -38,9 +42,7 @@ class Rocblas(CMakePackage):
depends_on('perl-file-which', type='build') depends_on('perl-file-which', type='build')
depends_on('py-pyyaml', type='build') depends_on('py-pyyaml', type='build')
depends_on('py-wheel', type='build') depends_on('py-wheel', type='build')
depends_on('py-msgpack', type='build')
# Tensile uses LLVM
depends_on('llvm-amdgpu')
resource(name='Tensile', resource(name='Tensile',
git='https://github.com/ROCmSoftwarePlatform/Tensile.git', git='https://github.com/ROCmSoftwarePlatform/Tensile.git',
@ -52,8 +54,13 @@ class Rocblas(CMakePackage):
commit='af71ea890a893e647bf2cf4571a90297d65689ca', commit='af71ea890a893e647bf2cf4571a90297d65689ca',
when='@3.7.0') when='@3.7.0')
resource(name='Tensile',
git='https://github.com/ROCmSoftwarePlatform/Tensile.git',
commit='9123205f9b5f95c96ff955695e942d2c3b321cbf',
when='@3.8.0')
# Status: https://github.com/ROCmSoftwarePlatform/Tensile/commit/a488f7dadba34f84b9658ba92ce9ec5a0615a087 # Status: https://github.com/ROCmSoftwarePlatform/Tensile/commit/a488f7dadba34f84b9658ba92ce9ec5a0615a087
# Not yet landed in 3.7.0. # Not yet landed in 3.7.0, nor 3.8.0.
patch('0001-Fix-compilation-error-with-StringRef-to-basic-string.patch') patch('0001-Fix-compilation-error-with-StringRef-to-basic-string.patch')
def setup_build_environment(self, env): def setup_build_environment(self, env):
@ -76,10 +83,12 @@ def cmake_args(self):
'-DTensile_ARCHITECTURE={0}'.format(archs), '-DTensile_ARCHITECTURE={0}'.format(archs),
'-DTensile_LOGIC=asm_full', '-DTensile_LOGIC=asm_full',
'-DTensile_CODE_OBJECT_VERSION=V3', '-DTensile_CODE_OBJECT_VERSION=V3',
'-DBUILD_WITH_TENSILE_HOST=OFF' '-DBUILD_WITH_TENSILE_HOST={0}'.format(
'ON' if '@3.7.0:' in self.spec else 'OFF'
)
] ]
if '@3.7.0' in self.spec: if '@3.7.0:' in self.spec:
args.append('-DTensile_LIBRARY_FORMAT=msgpack') args.append('-DTensile_LIBRARY_FORMAT=msgpack')
return args return args

5
var/spack/repos/builtin/packages/rocm-smi/package.py
View File

@ -30,5 +30,6 @@ def install(self, spec, prefix):
os.path.basename(self.spec['python'].command.path)), os.path.basename(self.spec['python'].command.path)),
'rocm_smi.py' 'rocm_smi.py'
) )
copy('rocm_smi.py', prefix) mkdir(prefix.bin)
symlink('rocm_smi.py', prefix.rocm_smi) copy('rocm_smi.py', prefix.bin)
symlink('rocm_smi.py', prefix.bin.rocm_smi)