import lit.llvm

config.llvm_tools_dir = r"@LLVM_TOOLS_DIR@"
config.akg_tools_dir = r"@AKG_MLIR_TOOLS_DIR@"
config.akg_test_build_dir = r"@CMAKE_BINARY_DIR@/tests"
config.lit_tools_dir = config.llvm_tools_dir
config.mlir_lib_dir = r"@MLIR_LIB_DIR@" or r"@LLVM_LIBRARY_DIR@"
config.llvm_shlib_ext = r"@SHLIBEXT@"

try:
    config.llvm_tools_dir = config.llvm_tools_dir % lit_config.params
except KeyError:
    e = sys.exc_info()[1]
    key, = e.args
    lit_config.fatal("unable to find %r parameter, use '--param=%s=VALUE'" % (key,key))

lit.llvm.initialize(lit_config, config)

# Let the main config do the real work.
lit_config.load_config(config, r"@CMAKE_SOURCE_DIR@/tests/lit.cfg.py")