From b2d6c634216a0ee2f0532372456ad389dd6fb427 Mon Sep 17 00:00:00 2001 From: eugeneswalker <38933153+eugeneswalker@users.noreply.github.com> Date: Sun, 1 Nov 2020 19:52:06 -0800 Subject: hip: setup run environments (#19637) * hip: rocminfo is a runtime requirement * hip: +setup_run_environment, +setup_dependent_run_environment * hip: run environment: get lib dir using libs.directories[0], not prefix.lib Co-authored-by: Adam J. Stewart Co-authored-by: Adam J. Stewart --- var/spack/repos/builtin/packages/hip/package.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/hip/package.py b/var/spack/repos/builtin/packages/hip/package.py index 7b93dcfda2..0487ebd182 100644 --- a/var/spack/repos/builtin/packages/hip/package.py +++ b/var/spack/repos/builtin/packages/hip/package.py @@ -32,7 +32,7 @@ class Hip(CMakePackage): depends_on('comgr@' + ver, type=('build', 'link', 'run'), when='@' + ver) depends_on('llvm-amdgpu@' + ver, type='build', when='@' + ver) depends_on('rocm-device-libs@' + ver, type=('build', 'link', 'run'), when='@' + ver) - depends_on('rocminfo@' + ver, type='build', when='@' + ver) + depends_on('rocminfo@' + ver, type=('build', 'run'), when='@' + ver) # Notice: most likely this will only be a hard dependency on 3.7.0 depends_on('numactl', when='@3.7.0:') @@ -48,6 +48,19 @@ class Hip(CMakePackage): # See https://github.com/ROCm-Developer-Tools/HIP/pull/2141 patch('0002-Fix-detection-of-HIP_CLANG_ROOT.patch', when='@3.5.0:') + def setup_run_environment(self, env): + env.set('ROCM_PATH', '') + env.set('HIP_COMPILER', 'clang') + env.set('HIP_PLATFORM', 'hcc') + env.set('HIP_CLANG_PATH', self.spec['llvm-amdgpu'].prefix.bin) + env.set('HSA_PATH', self.spec['hsa-rocr-dev'].prefix) + env.set('ROCMINFO_PATH', self.spec['rocminfo'].prefix) + env.set('DEVICE_LIB_PATH', + self.spec['rocm-device-libs'].libs.directories[0]) + + def setup_dependent_run_environment(self, env, dependent_spec): + self.setup_run_environment(env) + def get_rocm_prefix_info(self): # External packages in Spack do not currently contain dependency # information. External installations of hip therefore must compute -- cgit v1.2.3-60-g2f50