diff options
author | Sreenivasa Murthy Kolam <67086238+srekolam@users.noreply.github.com> | 2022-06-17 00:34:49 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-06-17 09:34:49 +0200 |
commit | 6ac31861327d07894f4793b5b8751712ecbfec57 (patch) | |
tree | f090bc7abc843a51b0cdf56b4d7f2e5211dee5cd /var/spack/repos/builtin/packages/miopen-tensile/package.py | |
parent | 466572dc14efcfd2054b52ee935dc809a39f1893 (diff) | |
download | spack-6ac31861327d07894f4793b5b8751712ecbfec57.tar.gz spack-6ac31861327d07894f4793b5b8751712ecbfec57.tar.bz2 spack-6ac31861327d07894f4793b5b8751712ecbfec57.tar.xz spack-6ac31861327d07894f4793b5b8751712ecbfec57.zip |
Support 'spack external find' for rocm packages (#30792)
Diffstat (limited to 'var/spack/repos/builtin/packages/miopen-tensile/package.py')
-rw-r--r-- | var/spack/repos/builtin/packages/miopen-tensile/package.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/miopen-tensile/package.py b/var/spack/repos/builtin/packages/miopen-tensile/package.py index 3a374e2856..f582d60df2 100644 --- a/var/spack/repos/builtin/packages/miopen-tensile/package.py +++ b/var/spack/repos/builtin/packages/miopen-tensile/package.py @@ -3,6 +3,7 @@ # # SPDX-License-Identifier: (Apache-2.0 OR MIT) +import re from spack.package import * @@ -16,6 +17,7 @@ class MiopenTensile(CMakePackage): url = "https://github.com/ROCmSoftwarePlatform/MIOpentensile/archive/rocm-5.0.0.tar.gz" maintainers = ['srekolam'] + libraries = ['libMIOpenTensile'] version('5.1.0', sha256='f1ae57bd4df8c154357b3f17caf0cfd5f80ba16ffff67bf6219a56f1eb5f897d') version('5.0.2', sha256='7b85a6a37d0905b0a3baa8361fd71a5a32ad90f3a562fd5e1af7e2ba68099fa6') @@ -54,6 +56,18 @@ class MiopenTensile(CMakePackage): def setup_build_environment(self, env): env.set('CXX', self.spec['hip'].hipcc) + @classmethod + def determine_version(cls, lib): + match = re.search(r'lib\S*\.so\.\d+\.\d+\.(\d)(\d\d)(\d\d)', + lib) + if match: + ver = '{0}.{1}.{2}'.format(int(match.group(1)), + int(match.group(2)), + int(match.group(3))) + else: + ver = None + return ver + def cmake_args(self): arch = self.spec.variants['tensile_architecture'].value tensile_path = join_path(self.stage.source_path, 'Tensile') |