mirror of https://github.com/llvm/torch-mlir
27 lines
972 B
Python
27 lines
972 B
Python
@LIT_SITE_CFG_IN_HEADER@
|
|
|
|
import sys
|
|
|
|
config.enable_bindings_python = @MLIR_ENABLE_BINDINGS_PYTHON@
|
|
config.torch_mlir_obj_root = "@TORCH_MLIR_BINARY_DIR@"
|
|
config.torch_mlir_python_packages_dir = "@TORCH_MLIR_PYTHON_PACKAGES_DIR@"
|
|
config.host_os = "@HOST_OS@"
|
|
config.llvm_src_root = "@LLVM_SOURCE_DIR@"
|
|
config.llvm_obj_root = "@LLVM_BINARY_DIR@"
|
|
config.llvm_tools_dir = "@LLVM_TOOLS_DIR@"
|
|
config.llvm_build_dir = "@CMAKE_BINARY_DIR@"
|
|
config.llvm_lib_dir = "@LLVM_LIBS_DIR@"
|
|
config.llvm_shlib_dir = "@SHLIBDIR@"
|
|
config.llvm_shlib_ext = "@SHLIBEXT@"
|
|
config.llvm_exe_ext = "@EXEEXT@"
|
|
config.lit_tools_dir = "@LLVM_LIT_TOOLS_DIR@"
|
|
config.python_executable = "@Python3_EXECUTABLE@"
|
|
config.enable_jit_ir_importer = @TORCH_MLIR_ENABLE_JIT_IR_IMPORTER@
|
|
config.enable_stablehlo = @TORCH_MLIR_ENABLE_STABLEHLO@
|
|
|
|
import lit.llvm
|
|
lit.llvm.initialize(lit_config, config)
|
|
|
|
# Let the main config do the real work.
|
|
lit_config.load_config(config, "@TORCH_MLIR_SOURCE_DIR@/test/lit.cfg.py")
|