HIP: Add RUNPATH compiled so files. (#510)

ROCm · Mar 19, 2024 · b395044 · b395044
1 parent d1564fa
commit b395044
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/triton/common/build.py b/python/triton/common/build.py
@@ -94,7 +94,7 @@ def _build(name, src, srcdir):
     if is_hip():
         ret = subprocess.check_call([
             cc, src, f"-I{hip_include_dir}", f"-I{py_include_dir}", f"-I{srcdir}", "-shared", "-fPIC",
-            f"-L{hip_lib_dir}", "-lamdhip64", "-o", so
+            f"-L{hip_lib_dir}", "-lamdhip64", f"-Wl,-rpath,{hip_lib_dir}", "-o", so
         ])
     else:
         cc_cmd = [