1*da0073e9SAndroid Build Coastguard Workerif(INTERN_BUILD_MOBILE) 2*da0073e9SAndroid Build Coastguard Worker caffe2_binary_target("speed_benchmark_torch.cc") 3*da0073e9SAndroid Build Coastguard Worker caffe2_binary_target("load_benchmark_torch.cc") 4*da0073e9SAndroid Build Coastguard Worker if(NOT BUILD_LITE_INTERPRETER) 5*da0073e9SAndroid Build Coastguard Worker caffe2_binary_target("compare_models_torch.cc") 6*da0073e9SAndroid Build Coastguard Worker endif() 7*da0073e9SAndroid Build Coastguard Worker return() 8*da0073e9SAndroid Build Coastguard Workerendif() 9*da0073e9SAndroid Build Coastguard Worker 10*da0073e9SAndroid Build Coastguard Workercaffe2_binary_target("parallel_info.cc") 11*da0073e9SAndroid Build Coastguard Workertarget_include_directories(parallel_info PUBLIC 12*da0073e9SAndroid Build Coastguard Worker ${CMAKE_BINARY_DIR}/aten/src) # provides "ATen/TypeExtendedInterface.h" to ATen.h 13*da0073e9SAndroid Build Coastguard Worker 14*da0073e9SAndroid Build Coastguard Workercaffe2_binary_target("record_function_benchmark.cc") 15*da0073e9SAndroid Build Coastguard Workertarget_include_directories(record_function_benchmark PUBLIC 16*da0073e9SAndroid Build Coastguard Worker ${CMAKE_BINARY_DIR}/aten/src) 17*da0073e9SAndroid Build Coastguard Worker 18*da0073e9SAndroid Build Coastguard Workercaffe2_binary_target("speed_benchmark_torch.cc") 19*da0073e9SAndroid Build Coastguard Workercaffe2_binary_target("compare_models_torch.cc") 20*da0073e9SAndroid Build Coastguard Worker 21*da0073e9SAndroid Build Coastguard Workerif(BUILD_TEST) 22*da0073e9SAndroid Build Coastguard Worker # Core overhead benchmark 23*da0073e9SAndroid Build Coastguard Worker caffe2_binary_target("core_overhead_benchmark.cc") 24*da0073e9SAndroid Build Coastguard Worker target_link_libraries(core_overhead_benchmark benchmark) 25*da0073e9SAndroid Build Coastguard Workerendif() 26*da0073e9SAndroid Build Coastguard Worker 27*da0073e9SAndroid Build Coastguard Workerif(USE_CUDA) 28*da0073e9SAndroid Build Coastguard Worker caffe2_binary_target("inspect_gpu.cc") 29*da0073e9SAndroid Build Coastguard Worker target_link_libraries(inspect_gpu ${CUDA_LIBRARIES}) 30*da0073e9SAndroid Build Coastguard Worker 31*da0073e9SAndroid Build Coastguard Workerendif() 32*da0073e9SAndroid Build Coastguard Worker 33*da0073e9SAndroid Build Coastguard Workerif(USE_ROCM) 34*da0073e9SAndroid Build Coastguard Worker caffe2_hip_binary_target("hip/inspect_gpu.cc") 35*da0073e9SAndroid Build Coastguard Worker 36*da0073e9SAndroid Build Coastguard Workerendif() 37*da0073e9SAndroid Build Coastguard Worker 38*da0073e9SAndroid Build Coastguard Workerif(USE_MPI) 39*da0073e9SAndroid Build Coastguard Worker caffe2_binary_target("run_plan_mpi.cc") 40*da0073e9SAndroid Build Coastguard Worker target_link_libraries(run_plan_mpi ${MPI_CXX_LIBRARIES}) 41*da0073e9SAndroid Build Coastguard Workerendif() 42*da0073e9SAndroid Build Coastguard Worker 43*da0073e9SAndroid Build Coastguard Worker 44*da0073e9SAndroid Build Coastguard Workercaffe2_binary_target("dump_operator_names.cc") 45*da0073e9SAndroid Build Coastguard Workercaffe2_binary_target("optimize_for_mobile.cc") 46*da0073e9SAndroid Build Coastguard Worker 47*da0073e9SAndroid Build Coastguard Workercaffe2_binary_target(aot_model_compiler "aot_model_compiler.cc") 48*da0073e9SAndroid Build Coastguard Workertarget_link_libraries(aot_model_compiler torch) 49