Skip to content

Commit ceb1091

Browse files
committed
Adding rpaths to mkl_info dict doesn't require mkl_root check
1 parent b3e9c3e commit ceb1091

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,13 +39,13 @@ def extensions():
3939
"include_dirs": [join(mkl_root, "include")],
4040
"library_dirs": [join(mkl_root, "lib"), join(mkl_root, "lib", "intel64")],
4141
"libraries": ["mkl_rt"],
42-
"rpaths": (
43-
["$ORIGIN/../..", "$ORIGIN/../../.."] if sys.platform != "win32" else []
44-
),
4542
}
4643
else:
4744
raise ValueError("MKLROOT environment variable not set.")
4845

46+
if sys.platform != "win32":
47+
mkl_info["rpaths"] = ["$ORIGIN/../..", "$ORIGIN/../../.."]
48+
4949
mkl_include_dirs = mkl_info.get("include_dirs", [])
5050
mkl_library_dirs = mkl_info.get("library_dirs", [])
5151
mkl_libraries = mkl_info.get("libraries", ["mkl_rt"])

0 commit comments

Comments
 (0)