torch-mlir/test/CMakeLists.txt

30 lines
837 B
CMake

llvm_canonicalize_cmake_booleans(
MLIR_ENABLE_BINDINGS_PYTHON
TORCH_MLIR_ENABLE_REFBACKEND
TORCH_MLIR_ENABLE_STABLEHLO
)
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 "Running the torch-mlir regression tests"
${CMAKE_CURRENT_BINARY_DIR}
DEPENDS ${TORCH_MLIR_TEST_DEPENDS}
)
set_target_properties(check-torch-mlir PROPERTIES FOLDER "Tests")
add_lit_testsuites(TORCH_MLIR ${CMAKE_CURRENT_SOURCE_DIR} DEPENDS ${TORCH_MLIR_TEST_DEPENDS})
add_subdirectory(CAPI)