From 6a958650c51d96e3addf9400366eb5333c714af7 Mon Sep 17 00:00:00 2001 From: Shrey Gupta Date: Thu, 18 Apr 2024 17:59:41 -0400 Subject: [PATCH] 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 0ef2e7d06a81..119a2c588c99 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: