Minor fix.

llvm · Jul 19, 2022 · 22adff8 · 22adff8
1 parent 55a1481
commit 22adff8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/test/Conversion/TorchToMhlo/lit.local.cfg b/test/Conversion/TorchToMhlo/lit.local.cfg
@@ -1,2 +1,2 @@
-if not config.add_mhlo_target:
-  config.unsupported = True
+if not config.enable_mhlo:
+  config.unsupported = True
diff --git a/test/lit.site.cfg.py.in b/test/lit.site.cfg.py.in
@@ -15,7 +15,7 @@ config.llvm_exe_ext = "@EXEEXT@"
 config.lit_tools_dir = "@LLVM_LIT_TOOLS_DIR@"
 config.python_executable = sys.executable
 config.enable_jit_ir_importer = @TORCH_MLIR_ENABLE_JIT_IR_IMPORTER@
-config.add_mhlo_target = @TORCH_MLIR_ENABLE_MHLO@
+config.enable_mhlo = @TORCH_MLIR_ENABLE_MHLO@
 
 import lit.llvm
 lit.llvm.initialize(lit_config, config)