Merge pull request #283 from ROCmSoftwarePlatform/default_path_fix_tr…

…iton_wheel Fix default rocm lib path
ROCm · Aug 4, 2023 · 34887ff · 34887ff
2 parents 9dc100a + b7b4b9b
commit 34887ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/triton/common/build.py b/python/triton/common/build.py
@@ -24,7 +24,7 @@ def libcuda_dirs():
 
 @functools.lru_cache()
 def rocm_path_dir():
-    default_path = os.path.join(os.path.dirname(__file__), "third_party", "rocm")
+    default_path = os.path.join(os.path.dirname(__file__), "..", "third_party", "rocm")
     # Check if include files have been populated locally.  If so, then we are 
     # most likely in a whl installation and he rest of our libraries should be here
     if (os.path.exists(default_path+"/include/hip/hip_runtime.h")):