diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/rccl/package.py')
| -rw-r--r-- | var/spack/repos/builtin/packages/rccl/package.py | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/rccl/package.py b/var/spack/repos/builtin/packages/rccl/package.py index 07825d3a2c..989723454d 100644 --- a/var/spack/repos/builtin/packages/rccl/package.py +++ b/var/spack/repos/builtin/packages/rccl/package.py @@ -21,13 +21,24 @@ class Rccl(CMakePackage): version('3.7.0', sha256='8273878ff71aac2e7adf5cc8562d2933034c6c6b3652f88fbe3cd4f2691036e3') version('3.5.0', sha256='290b57a66758dce47d0bfff3f5f8317df24764e858af67f60ddcdcadb9337253') + patch('0001-Fix-numactl-path-issue.patch', when='@3.7.0:') + depends_on('cmake@3:', type='build') for ver in ['3.5.0', '3.7.0', '3.8.0']: depends_on('rocm-cmake@' + ver, type='build', when='@' + ver) - depends_on('hip@' + ver, type='build', when='@' + ver) + depends_on('hip@' + ver, type=('build', 'run'), when='@' + ver) depends_on('rocm-device-libs@' + ver, type='build', when='@' + ver) depends_on('comgr@' + ver, type='build', when='@' + ver) depends_on('hsa-rocr-dev@' + ver, type='build', when='@' + ver) + if ver in ['3.7.0', '3.8.0']: + depends_on('numactl@2.0.12', type=('build', 'link'), when='@' + ver) def setup_build_environment(self, env): env.set('CXX', self.spec['hip'].hipcc) + + def cmake_args(self): + args = [] + if '@3.7.0:' in self.spec: + numactl_prefix = self.spec['numactl'].prefix + args.append('-DNUMACTL_DIR={0}'.format(numactl_prefix)) + return args |