diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/rccl/package.py')
| -rw-r--r-- | var/spack/repos/builtin/packages/rccl/package.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/rccl/package.py b/var/spack/repos/builtin/packages/rccl/package.py index 4544fafedb..51c23b8143 100644 --- a/var/spack/repos/builtin/packages/rccl/package.py +++ b/var/spack/repos/builtin/packages/rccl/package.py @@ -3,6 +3,8 @@ # # SPDX-License-Identifier: (Apache-2.0 OR MIT) +import re + from spack import * @@ -17,6 +19,7 @@ class Rccl(CMakePackage): url = "https://github.com/ROCmSoftwarePlatform/rccl/archive/rocm-5.0.0.tar.gz" maintainers = ['srekolam', 'arjun-raj-kuppala'] + libraries = ['librccl.so'] version('5.0.2', sha256='a2377ad2332b93d3443a8ee74f4dd9f965ae8cbbfad473f8f57ca17905389a39') version('5.0.0', sha256='80eb70243f11b80e215458a67c278cd5a655f6e486289962b92ba3504e50af5c') @@ -53,6 +56,18 @@ class Rccl(CMakePackage): for ver in ['4.5.0', '4.5.2', '5.0.0', '5.0.2']: depends_on('rocm-smi-lib@' + ver, when='@' + ver) + @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 setup_build_environment(self, env): env.set('CXX', self.spec['hip'].hipcc) |