Skip to content

Commit 2b813ec

Browse files
[Upd] Enable lld search to include /opt/rocm/llvm/bin for rocm (#16540)
1 parent 059f629 commit 2b813ec

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/tvm/contrib/rocm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def find_lld(required=True):
5252
if major is not None:
5353
lld_list += [f"ld.lld-{major}.0"]
5454
lld_list += [f"ld.lld-{major}"]
55-
lld_list += ["ld.lld"]
55+
lld_list += ["ld.lld", "/opt/rocm/llvm/bin"]
5656
valid_list = [utils.which(x) for x in lld_list]
5757
valid_list = [x for x in valid_list if x]
5858
if not valid_list and required:

0 commit comments

Comments
 (0)