2023-01-17 00:25:02 +08:00
|
|
|
add_llvm_executable(torch-mlir-capi-torch-test torch.c)
|
|
|
|
|
llvm_update_compile_flags(torch-mlir-capi-torch-test)
|
|
|
|
|
target_link_libraries(
|
|
|
|
|
torch-mlir-capi-torch-test
|
|
|
|
|
PRIVATE
|
|
|
|
|
MLIRCAPIIR
|
|
|
|
|
MLIRCAPIRegisterEverything
|
|
|
|
|
TorchMLIRCAPI
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
add_lit_testsuite(check-torch-mlir-capi "Running the torch-mlir CAPI tests"
|
|
|
|
|
${CMAKE_CURRENT_BINARY_DIR}
|
2023-07-25 16:11:55 +08:00
|
|
|
DEPENDS torch-mlir-capi-torch-test FileCheck
|
2023-01-17 00:25:02 +08:00
|
|
|
)
|
2023-07-25 16:11:55 +08:00
|
|
|
set_target_properties(check-torch-mlir-capi PROPERTIES FOLDER "Tests")
|
