From 915cbb9471bfda72f76d77b6c2d572ff345a1acc Mon Sep 17 00:00:00 2001 From: Gabriele Oliaro Date: Fri, 14 Jul 2023 22:35:47 -0400 Subject: [PATCH] fix --- python/flexflow/flexflow_python | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/flexflow/flexflow_python b/python/flexflow/flexflow_python index f327b168e0..471a0d07fb 100644 --- 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 "$@" \ No newline at end of file