blob: 5d37f8bb99d9a1c6aa3812ebca48f20f2a0df029 [file] [log] [blame]
set(MOBILE_NNC_TEST_ROOT ${TORCH_ROOT}/test/mobile/nnc)
set(MOBILE_NNC_TEST_SRCS
${MOBILE_NNC_TEST_ROOT}/test_context.cpp
${MOBILE_NNC_TEST_ROOT}/test_nnc_backend.cpp
${MOBILE_NNC_TEST_ROOT}/test_registry.cpp
)
add_executable(test_mobile_nnc
${TORCH_ROOT}/test/cpp/lite_interpreter_runtime/main.cpp
${MOBILE_NNC_TEST_SRCS}
)
target_link_libraries(test_mobile_nnc PRIVATE torch gtest)
target_include_directories(test_mobile_nnc PRIVATE ${ATen_CPU_INCLUDE})
target_compile_definitions(test_mobile_nnc PRIVATE USE_GTEST)
if(INSTALL_TEST)
install(TARGETS test_mobile_nnc DESTINATION bin)
# Install PDB files for MSVC builds
if(MSVC AND BUILD_SHARED_LIBS)
install(FILES $<TARGET_PDB_FILE:test_mobile_nnc> DESTINATION bin OPTIONAL)
endif()
endif()