llvm-amdgpu: Updating LD_LIBRARY_PATH w.r.t new prefix path (#45940)

* Updating LD_LIBRARY_PATH w.r.t new prefix path

* Updating hsa external path for 6.x
This commit is contained in:
renjithravindrankannath 2024-08-28 11:26:05 -07:00 committed by GitHub
parent ec0a57cba4
commit ee27dc5d45
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 2 deletions

View File

@ -359,6 +359,9 @@ def get_paths(self):
if self.spec.satisfies("@5.7:"):
paths["hip-path"] = rocm_prefix
if self.spec.satisfies("@6.0:"):
paths["hsa-rocr-dev"] = rocm_prefix
else:
paths = {
"hip-path": self.spec.prefix,

View File

@ -274,12 +274,12 @@ def cmake_args(self):
# Make sure that the compiler paths are in the LD_LIBRARY_PATH
def setup_run_environment(self, env):
llvm_amdgpu_home = self.spec["llvm-amdgpu"].prefix
env.prepend_path("LD_LIBRARY_PATH", llvm_amdgpu_home + "/llvm/lib")
env.prepend_path("LD_LIBRARY_PATH", llvm_amdgpu_home + "/lib")
# Make sure that the compiler paths are in the LD_LIBRARY_PATH
def setup_dependent_run_environment(self, env, dependent_spec):
llvm_amdgpu_home = self.spec["llvm-amdgpu"].prefix
env.prepend_path("LD_LIBRARY_PATH", llvm_amdgpu_home + "/llvm/lib")
env.prepend_path("LD_LIBRARY_PATH", llvm_amdgpu_home + "/lib")
# Required for enabling asan on dependent packages
def setup_dependent_build_environment(self, env, dependent_spec):