fix

flexflow · Jul 15, 2023 · 915cbb9 · 915cbb9
1 parent 323ba82
commit 915cbb9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/flexflow/flexflow_python b/python/flexflow/flexflow_python
@@ -3,7 +3,9 @@
 # This file runs the flexflow_python interpreter when installing FlexFlow via pip. This file can be ignored when installing FlexFlow using cmake/make
 
 python_packages=$(python -c "from distutils import sysconfig; print(sysconfig.get_python_lib(plat_specific=False,standard_lib=False))")
+pylib_path="$("$python_packages"/flexflow/findpylib.py)"
+pylib_dir="$(dirname "$pylib_path")"
 export PATH="${python_packages}/flexflow/bin:${PATH}"
-export LD_LIBRARY_PATH="${python_packages}/flexflow/lib:${PATH}"
+export LD_LIBRARY_PATH="${python_packages}/flexflow/lib:${pylib_dir}:${PATH}"
 
 legion_python "$@"