1cmake_minimum_required(VERSION 3.1) 2 3set(TORCH_ROOT ${CMAKE_CURRENT_LIST_DIR}/../../..) 4set(TEST_ROOT ${TORCH_ROOT}/test/mobile/lightweight_dispatch) 5 6add_executable(test_codegen_unboxing 7 ${TEST_ROOT}/test_lightweight_dispatch.cpp 8 ${TEST_ROOT}/test_codegen_unboxing.cpp 9) 10 11target_include_directories(test_codegen_unboxing PRIVATE ${ATen_CPU_INCLUDE}) 12 13target_compile_definitions(test_codegen_unboxing PRIVATE USE_GTEST) 14 15set(TEST_UNBOXING_DEPENDENCIES torch gtest) 16 17target_link_libraries(test_codegen_unboxing PRIVATE 18 ${TEST_UNBOXING_DEPENDENCIES} 19) 20 21if(INSTALL_TEST) 22 install(TARGETS test_codegen_unboxing DESTINATION bin) 23endif() 24