fix compile sh path

synnada-ai · Mar 4, 2025 · 21e0a09 · 21e0a09
1 parent 2494482
commit 21e0a09
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -22,15 +22,25 @@
 class CustomBuildExt(build_ext):
     def run(self):
         shell = os.getenv("SHELL", "sh")
-        script_path = os.path.join(
+        path_1 = os.path.join(
             os.path.dirname(__file__),
             "mithril",
             "cores",
             "c",
             "raw_c",
             "compile.sh",
         )
-        subprocess.check_call([shell, script_path])
+        path_2 = os.path.join(
+            os.path.dirname(__file__),
+            "mithril",
+            "cores",
+            "c",
+            "ggml",
+            "compile.sh",
+        )
+
+        subprocess.check_call([shell, path_1])
+        subprocess.check_call([shell, path_2])
         # Continue with the normal build
         super().run()