diff options
Diffstat (limited to 'mlir/test/lit.cfg.py')
| -rw-r--r-- | mlir/test/lit.cfg.py | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/mlir/test/lit.cfg.py b/mlir/test/lit.cfg.py index feaf5fb..f99c24d 100644 --- a/mlir/test/lit.cfg.py +++ b/mlir/test/lit.cfg.py @@ -29,6 +29,7 @@ lit_shell_env = os.environ.get("LIT_USE_INTERNAL_SHELL") if lit_shell_env: use_lit_shell = lit.util.pythonize_bool(lit_shell_env) +# Set the test format based on shell configuration config.test_format = lit.formats.ShTest(execute_external=not use_lit_shell) # suffixes: A list of file extensions to treat as test files. @@ -181,15 +182,16 @@ config.excludes = [ ] # Tweak the PATH to include the tools dir. -llvm_config.with_environment("PATH", config.mlir_tools_dir, append_path=True) -llvm_config.with_environment("PATH", config.llvm_tools_dir, append_path=True) +tool_dirs = [config.mlir_tools_dir, config.llvm_tools_dir, config.lit_tools_dir] +for dirs in tool_dirs: + llvm_config.with_environment("PATH", dirs, append_path=True) -tool_dirs = [config.mlir_tools_dir, config.llvm_tools_dir] tools = [ "mlir-tblgen", "mlir-translate", "mlir-lsp-server", "mlir-capi-execution-engine-test", + "mlir-capi-global-constructors-test", "mlir-capi-ir-test", "mlir-capi-irdl-test", "mlir-capi-llvm-test", @@ -224,6 +226,9 @@ if config.enable_cuda_runner: if config.enable_sycl_runner: tools.extend([add_runtime("mlir_sycl_runtime")]) +if config.enable_levelzero_runner: + tools.extend([add_runtime("mlir_levelzero_runtime")]) + if config.enable_spirv_cpu_runner: tools.extend([add_runtime("mlir_spirv_cpu_runtime")]) @@ -375,3 +380,6 @@ if config.run_rocm_tests: if config.arm_emulator_executable: config.available_features.add("arm-emulator") + +if sys.version_info >= (3, 11): + config.available_features.add("python-ge-311") |