fixed setup for standalone install

hariharan-devarajan · Aug 10, 2023 · 90d3a0d · 90d3a0d
1 parent 4620427
commit 90d3a0d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -40,6 +40,9 @@ def build_extension(self, ext: CMakeExtension) -> None:
         if "DLIO_PROFILER_DIR" in os.environ:
             install_prefix = os.environ['DLIO_PROFILER_DIR']
         cmake_args += [f"-DCMAKE_INSTALL_PREFIX={install_prefix}"]
+        if "DLIO_PYTHON_SITE" in os.environ:
+            site = os.environ['DLIO_PYTHON_SITE']
+            cmake_args += [f"-DDLIO_PYTHON_SITE={site}"]
         project_dir = Path.cwd()
         dependency_file = open(f"{project_dir}/dependency/cpp.requirements.txt", 'r')
         dependencies = dependency_file.readlines()