Update hip_backend to use libhsa-runtime for arch info, (#411)

brings in path changes for pytorch triton wheels

Co-authored-by: jayfurmanek <Jason.Furmanek@amd.com>
This commit is contained in:
Jack Taylor
2023-12-21 15:40:57 +00:00
committed by GitHub
parent 0248bdb29d
commit 1e2fd0dd1a
8 changed files with 21 additions and 23 deletions

View File

@@ -232,7 +232,7 @@ std::string generate_hsaco(llvm::Module *module, const std::string &triple,
.parent_path()
.parent_path() /
"triton" / "third_party" /
"rocm" / "llvm" / "bin" / "ld.lld";
"hip" / "llvm" / "bin" / "ld.lld";
std::string lld_path = compiletime_path.string();
if (!std::filesystem::exists(lld_path)) {
std::string rocm_path = ::triton::tools::getenv("ROCM_PATH");