(tvm) branch main updated: [Upd] Enable lld search to include /opt/rocm/llvm/bin for rocm (#16540)

[tq...@apache.org]

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 2b813ec04f [Upd] Enable lld search to include /opt/rocm/llvm/bin for rocm (#16540)
2b813ec04f is described below

commit 2b813ec04f1a65d6a21b99c7435096d2f8c77b4c
Author: Shrey Gupta <51...@users.noreply.github.com>
AuthorDate: Wed Feb 14 08:40:02 2024 -0500

    [Upd] Enable lld search to include /opt/rocm/llvm/bin for rocm (#16540)
---
 python/tvm/contrib/rocm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/contrib/rocm.py b/python/tvm/contrib/rocm.py
index fb50c71e22..0ef2e7d06a 100644
--- a/python/tvm/contrib/rocm.py
+++ b/python/tvm/contrib/rocm.py
@@ -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: