From 22adff85ba99c245acb870a01399e0340bad12bd Mon Sep 17 00:00:00 2001 From: Ziheng Jiang Date: Wed, 20 Jul 2022 04:14:56 +0800 Subject: [PATCH] Minor fix. --- test/Conversion/TorchToMhlo/lit.local.cfg | 4 ++-- test/lit.site.cfg.py.in | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/test/Conversion/TorchToMhlo/lit.local.cfg b/test/Conversion/TorchToMhlo/lit.local.cfg index 584a8c8a27a2..829a5662f6e6 100644 --- 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 \ No newline at end of file +if not config.enable_mhlo: + config.unsupported = True diff --git a/test/lit.site.cfg.py.in b/test/lit.site.cfg.py.in index fb275ff34068..94c93a05c898 100644 --- 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)