Load system PYTHONPATH during LLVM lit tests

llvm · Jun 30, 2022 · cfe0d56 · cfe0d56
1 parent 7d1b37f
commit cfe0d56
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/test/lit.cfg.py b/test/lit.cfg.py
@@ -67,6 +67,8 @@
 
 if config.enable_bindings_python:
   llvm_config.with_environment('PYTHONPATH', [
-      os.path.join(config.torch_mlir_python_packages_dir, 'torch_mlir'),
-  ],
-                               append_path=True)
+      os.pathsep.join([
+          os.path.join(config.torch_mlir_python_packages_dir, 'torch_mlir'),
+          os.environ['PYTHONPATH'],
+      ]),
+  ], append_path=True)