Switching from chrpath to patchelf

Author: spollok

python/.gitignore

@@ -11,7 +11,6 @@ __pycache__/
 build/
 develop-eggs/
 dist/
-dist_old/
 downloads/
 eggs/
 .eggs/

python/build_env_linux.yml

@@ -102,4 +102,5 @@ dependencies:
       - ninja==1.11.1.3
       - numpy==2.2.3
       - packaging==24.2
+      - patchelf==0.17.2.1
       - pyproject-hooks==1.2.0

python/requirements-dev.txt

@@ -4,6 +4,7 @@ meson==1.7.0
 ninja==1.11.1.3
 numpy==2.2.3
 packaging==24.2
+patchelf==0.17.2.1
 pyproject_hooks==1.2.0
 setuptools==75.8.2
 wheel==0.45.1

python/scripts/dist_pypi.py

@@ -58,12 +58,24 @@ def main(py_versions):
                 subprocess.run(["rm", lib_path])
             lib_path = f"src/magtense/lib/magtensesource.{py_lib}-{arch}.{suffix}"
             if platform == "linux":
-                # subprocess.run(
-                #     ["patchelf", "--set-rpath", "$ORIGIN/../../../../../lib", lib_path]
-                # )
-                subprocess.run(
-                    ["chrpath", "-r", "$ORIGIN/../../../../../lib", lib_path]
-                )
+                if cuda == "cpu":
+                    subprocess.run(
+                        [
+                            "patchelf",
+                            "--set-rpath",
+                            "$ORIGIN/../../../../../lib",
+                            lib_path,
+                        ]
+                    )
+                elif cuda == "cu12":
+                    subprocess.run(
+                        [
+                            "patchelf",
+                            "--set-rpath",
+                            "$ORIGIN/../../../../../lib:$ORIGIN/../../nvidia/cublas/lib/:$ORIGIN/../../nvidia/cuda_runtime/lib/:$ORIGIN/../../nvidia/cusparse/lib/",
+                            lib_path,
+                        ]
+                    )
             subprocess.run(["cp", f"requirements-{cuda}.txt", "requirements.txt"])
             subprocess.run(["python3", "-m", "build", "--wheel"])
             subprocess.run(