#441: Enable standalone ttrt build (#444)

tenstorrent · Aug 19, 2024 · 8a7c7b0 · 8a7c7b0
1 parent 942bd40
commit 8a7c7b0
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/runtime/tools/python/setup.py b/runtime/tools/python/setup.py
@@ -6,6 +6,7 @@
 from pybind11.setup_helpers import Pybind11Extension, build_ext
 from setuptools import setup
 import shutil
+import subprocess
 
 TTMLIR_VERSION_MAJOR = os.getenv("TTMLIR_VERSION_MAJOR", "0")
 TTMLIR_VERSION_MINOR = os.getenv("TTMLIR_VERSION_MINOR", "0")
@@ -62,6 +63,24 @@
             f"{metallibdir}/{dylib}",
             f"{src_dir}/build/runtime/tools/python/ttrt/runtime",
         )
+        command = [
+            "patchelf",
+            "--set-rpath",
+            "$ORIGIN",
+            f"{src_dir}/build/runtime/tools/python/ttrt/runtime/{dylib}",
+        ]
+
+        try:
+            result = subprocess.run(
+                command,
+                check=True,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True,
+            )
+        except subprocess.CalledProcessError as e:
+            print(f"Command failed with return code {e.returncode}")
+
     ext_modules.append(
         Pybind11Extension(
             "ttrt.runtime._C",