20 lines
628 B
Python
20 lines
628 B
Python
|
|
import lit.llvm
|
|
|
|
if '@BUILD_SHARED_LIBS@' == 'ON':
|
|
config.environment['LD_PRELOAD'] = "@ONNX_MLIR_LD_PRELOAD_onnx-mlir-opt@"
|
|
|
|
config.llvm_tools_dir = "@MLIR_TOOLS_DIR@"
|
|
config.mlir_obj_root = "@LLVM_PROJ_BUILD@"
|
|
config.mlir_tools_dir = "@MLIR_TOOLS_DIR@"
|
|
config.suffixes = ['.mlir']
|
|
|
|
config.onnx_mlir_tools_dir = "@ONNX_MLIR_TOOLS_DIR@"
|
|
config.onnx_mlir_test_src_dir = "@ONNX_MLIR_LIT_TEST_SRC_DIR@"
|
|
config.onnx_mlir_test_build_dir = "@ONNX_MLIR_LIT_TEST_BUILD_DIR@"
|
|
|
|
lit.llvm.initialize(lit_config, config)
|
|
|
|
# Let the main config do the real work.
|
|
lit_config.load_config(config, "@ONNX_MLIR_LIT_TEST_SRC_DIR@/lit.cfg.py")
|