From 323ba825274df6279d9355bd9e769a1c06f5d24a Mon Sep 17 00:00:00 2001
From: Gabriele Oliaro <goliaro@cs.cmu.edu>
Date: Fri, 14 Jul 2023 22:05:22 -0400
Subject: [PATCH] fix

---
 python/flexflow/flexflow_python | 1 +
 setup.py                        | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/flexflow/flexflow_python b/python/flexflow/flexflow_python
index ac7c5dd827..f327b168e0 100644
--- a/python/flexflow/flexflow_python
+++ b/python/flexflow/flexflow_python
@@ -4,5 +4,6 @@
 
 python_packages=$(python -c "from distutils import sysconfig; print(sysconfig.get_python_lib(plat_specific=False,standard_lib=False))")
 export PATH="${python_packages}/flexflow/bin:${PATH}"
+export LD_LIBRARY_PATH="${python_packages}/flexflow/lib:${PATH}"
 
 legion_python "$@"
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 6f63130cd6..4564657d46 100644
--- a/setup.py
+++ b/setup.py
@@ -39,7 +39,7 @@
         "pybind11",
         "ninja",
     ],
-    scripts=['flexflow/flexflow_python'],
+    scripts=['python/flexflow/flexflow_python'],
     ext_modules=[
         CMakeExtension(
             name="flexflow",