This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 36efa36f53 [Upd] Fixed lld search in rocm (#16907)
36efa36f53 is described below
commit 36efa36f53f4ad9f302ece4208e5b8296c86c8bb
Author: Shrey Gupta <[email protected]>
AuthorDate: Fri Apr 19 04:41:05 2024 -0400
[Upd] Fixed lld search in rocm (#16907)
fixed lld search
---
python/tvm/contrib/rocm.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/python/tvm/contrib/rocm.py b/python/tvm/contrib/rocm.py
index 0ef2e7d06a..119a2c588c 100644
--- a/python/tvm/contrib/rocm.py
+++ b/python/tvm/contrib/rocm.py
@@ -52,7 +52,8 @@ def find_lld(required=True):
if major is not None:
lld_list += [f"ld.lld-{major}.0"]
lld_list += [f"ld.lld-{major}"]
- lld_list += ["ld.lld", "/opt/rocm/llvm/bin"]
+ lld_list += ["ld.lld"]
+ lld_list += [f"/opt/rocm/llvm/bin/{x}" for x in lld_list]
valid_list = [utils.which(x) for x in lld_list]
valid_list = [x for x in valid_list if x]
if not valid_list and required:
