[Mlir-commits] [mlir] 8c212a3 - [mlir][python] allow passing in PYTHONPATH to lit tests (#89296)

[llvmlistbot at llvm.org]

Author: Maksim Levental
Date: 2024-04-18T18:06:11-05:00
New Revision: 8c212a3f41e2414cbdea274fd5fd4188893df72c

URL: https://github.com/llvm/llvm-project/commit/8c212a3f41e2414cbdea274fd5fd4188893df72c
DIFF: https://github.com/llvm/llvm-project/commit/8c212a3f41e2414cbdea274fd5fd4188893df72c.diff

LOG: [mlir][python] allow passing in PYTHONPATH to lit tests (#89296)

Added: 
    

Modified: 
    mlir/test/lit.cfg.py

Removed: 
    


################################################################################
diff  --git a/mlir/test/lit.cfg.py b/mlir/test/lit.cfg.py
index 7636ef30c2d3ef..4740e7d137e8d3 100644
--- a/mlir/test/lit.cfg.py
+++ b/mlir/test/lit.cfg.py
@@ -204,6 +204,7 @@ def add_runtime(name):
 # binaries come from the build tree. This should be unified to the build tree
 # by copying/linking sources to build.
 if config.enable_bindings_python:
+    config.environment["PYTHONPATH"] = os.getenv("MLIR_LIT_PYTHONPATH", "")
     llvm_config.with_environment(
         "PYTHONPATH",
         [