1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
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)
add_executable(aot_model_compiler_test
${TORCH_ROOT}/binaries/aot_model_compiler.cc
)
target_link_libraries(aot_model_compiler_test PRIVATE torch)
target_include_directories(aot_model_compiler_test PRIVATE ${ATen_CPU_INCLUDE})
if(INSTALL_TEST)
install(TARGETS test_mobile_nnc DESTINATION bin)
install(TARGETS aot_model_compiler_test 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()
|
