Skip to content

Commit

Permalink
[Upd] Enable lld search to include /opt/rocm/llvm/bin for rocm (#16540)
Browse files Browse the repository at this point in the history
  • Loading branch information
shreygupta2809 committed Feb 14, 2024
1 parent 059f629 commit 2b813ec
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/tvm/contrib/rocm.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ 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"]
lld_list += ["ld.lld", "/opt/rocm/llvm/bin"]
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:
Expand Down

0 comments on commit 2b813ec

Please sign in to comment.