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

makslevental · web-flow · commit 8c212a3f41e2 · 2024-04-18T18:06:11.000-05:00
diff --git 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",
         [

Original file line number	Diff line number	Diff line change
`@@ -204,6 +204,7 @@ def add_runtime(name):`
`204`	`204`	`# binaries come from the build tree. This should be unified to the build tree`
`205`	`205`	`# by copying/linking sources to build.`
`206`	`206`	`if config.enable_bindings_python:`
	`207`	`+ config.environment["PYTHONPATH"] = os.getenv("MLIR_LIT_PYTHONPATH", "")`
`207`	`208`	`llvm_config.with_environment(`
`208`	`209`	`"PYTHONPATH",`
`209`	`210`	`[`