mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:07:10 +08:00
c55af2b567
As it's unbuildable right now, as headers it depends on are gone Fixes https://github.com/pytorch/pytorch/issues/146042 Pull Request resolved: https://github.com/pytorch/pytorch/pull/146105 Approved by: https://github.com/atalman, https://github.com/seemethere
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)
|