32 lines
1.0 KiB
CMake
32 lines
1.0 KiB
CMake
if(INTERN_BUILD_MOBILE)
|
|
caffe2_binary_target("speed_benchmark_torch.cc")
|
|
caffe2_binary_target("load_benchmark_torch.cc")
|
|
if(NOT BUILD_LITE_INTERPRETER)
|
|
caffe2_binary_target("compare_models_torch.cc")
|
|
endif()
|
|
return()
|
|
endif()
|
|
|
|
caffe2_binary_target("parallel_info.cc")
|
|
target_include_directories(parallel_info PUBLIC
|
|
${CMAKE_BINARY_DIR}/aten/src) # provides "ATen/TypeExtendedInterface.h" to ATen.h
|
|
|
|
caffe2_binary_target("record_function_benchmark.cc")
|
|
target_include_directories(record_function_benchmark PUBLIC
|
|
${CMAKE_BINARY_DIR}/aten/src)
|
|
|
|
caffe2_binary_target("speed_benchmark_torch.cc")
|
|
caffe2_binary_target("compare_models_torch.cc")
|
|
|
|
if(BUILD_TEST)
|
|
# Core overhead benchmark
|
|
caffe2_binary_target("core_overhead_benchmark.cc")
|
|
target_link_libraries(core_overhead_benchmark benchmark)
|
|
endif()
|
|
|
|
caffe2_binary_target("dump_operator_names.cc")
|
|
caffe2_binary_target("optimize_for_mobile.cc")
|
|
|
|
caffe2_binary_target(aot_model_compiler "aot_model_compiler.cc")
|
|
target_link_libraries(aot_model_compiler torch)
|