mirror of https://github.com/llvm/torch-mlir
27 lines
800 B
CMake
27 lines
800 B
CMake
|
llvm_canonicalize_cmake_booleans(
|
||
|
|
MLIR_ENABLE_BINDINGS_PYTHON
|
||
|
|
TORCH_MLIR_ENABLE_JIT_IR_IMPORTER
|
||
|
|
)
|
||
|
|
|
||
|
|
configure_lit_site_cfg(
|
||
|
|
${CMAKE_CURRENT_SOURCE_DIR}/lit.site.cfg.py.in
|
||
|
|
${CMAKE_CURRENT_BINARY_DIR}/lit.site.cfg.py
|
||
|
|
MAIN_CONFIG
|
||
|
|
${CMAKE_CURRENT_SOURCE_DIR}/lit.cfg.py
|
||
|
|
)
|
||
|
|
|
||
|
|
set(TORCH_MLIR_TEST_DEPENDS
|
||
|
|
FileCheck count not
|
||
|
|
TorchMLIRPythonModules
|
||
|
|
torch-mlir-opt
|
||
|
|
torch-mlir-capi-torch-test
|
||
|
|
)
|
||
|
|
|
||
|
|
add_lit_testsuite(check-torch-mlir-pt1 "Running the torch-mlir PT1 regression tests"
|
||
|
|
${CMAKE_CURRENT_BINARY_DIR}
|
||
|
|
DEPENDS ${TORCH_MLIR_TEST_DEPENDS}
|
||
|
|
)
|
||
|
|
set_target_properties(check-torch-mlir-pt1 PROPERTIES FOLDER "Tests")
|
||
|
|
|
||
|
|
add_lit_testsuites(TORCH_MLIR ${CMAKE_CURRENT_SOURCE_DIR} DEPENDS ${TORCH_MLIR_TEST_DEPENDS})
|