From cfe0d56c1ccaf2d45e548dc4945a8d8db8de94ac Mon Sep 17 00:00:00 2001
From: Henry Tu <henry.tu@cerebras.net>
Date: Fri, 11 Feb 2022 10:20:04 -0500
Subject: [PATCH] Load system PYTHONPATH during LLVM lit tests

---
 test/lit.cfg.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/test/lit.cfg.py b/test/lit.cfg.py
index 5a7ba132f3b..6b841c96815 100644
--- 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)