torch-mlir/frontends/pytorch/test/CMakeLists.txt

24 lines
684 B
CMake

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(TEST_DEPENDS
FileCheck count not
npcomp-opt
TorchMLIRTorchPlugin
TorchMLIRPluginPythonModules
NPCOMPPythonModules
)
add_lit_testsuite(check-frontends-pytorch "Running the frontends-pytorch regression tests"
${CMAKE_CURRENT_BINARY_DIR}
DEPENDS ${TEST_DEPENDS}
)
set_target_properties(check-frontends-pytorch PROPERTIES FOLDER "Tests")
add_lit_testsuites(FRONTENDS_PYTORCH ${CMAKE_CURRENT_SOURCE_DIR} DEPENDS ${TEST_DEPENDS})
add_dependencies(check-npcomp-all check-frontends-pytorch)