[Dependency] Fix wrong so path returned in get_env(). (#432)

intel · Jun 5, 2024 · a2d33c6 · a2d33c6
1 parent e23c8f8
commit a2d33c6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/xfastertransformer/env.py b/src/xfastertransformer/env.py
@@ -14,6 +14,8 @@
 # ============================================================================
 import os
 def get_env():
-    libiomp_path = os.path.dirname(os.path.abspath(__file__)) + "/libxft_comm_helper.so"
-    env = f"LD_PRELOAD={libiomp_path}:$LD_PRELOAD"
+    libiomp_path = os.path.dirname(os.path.abspath(__file__)) + "/libiomp5.so"
+    env = f"LD_PRELOAD={libiomp_path}"
+    if os.getenv("LD_PRELOAD"):
+        env = f"{env}:{os.getenv('LD_PRELOAD')}"
     return env