mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
2ed17e0b1e
This PR removed some binaries using deleted or to be deleted Caffe2 functions. Pull Request resolved: https://github.com/pytorch/pytorch/pull/125885 Approved by: https://github.com/r-barnes, https://github.com/Chillee
59 lines
1.6 KiB
CMake
59 lines
1.6 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()
|
|
|
|
if(BUILD_CAFFE2)
|
|
caffe2_binary_target("at_launch_benchmark.cc")
|
|
target_include_directories(at_launch_benchmark PUBLIC
|
|
${CMAKE_BINARY_DIR}/aten/src)
|
|
|
|
caffe2_binary_target("intra_inter_benchmark.cc")
|
|
target_include_directories(intra_inter_benchmark PUBLIC
|
|
${CMAKE_BINARY_DIR}/aten/src)
|
|
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()
|
|
|
|
if(USE_CUDA)
|
|
caffe2_binary_target("inspect_gpu.cc")
|
|
target_link_libraries(inspect_gpu ${CUDA_LIBRARIES})
|
|
|
|
endif()
|
|
|
|
if(USE_ROCM)
|
|
caffe2_hip_binary_target("hip/inspect_gpu.cc")
|
|
|
|
endif()
|
|
|
|
if(USE_MPI)
|
|
caffe2_binary_target("run_plan_mpi.cc")
|
|
target_link_libraries(run_plan_mpi ${MPI_CXX_LIBRARIES})
|
|
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)
|