[cmake] Uncomment binaries (pytorch#71157)

Summary: Pull Request resolved: pytorch#71157 Test Plan: Imported from OSS Reviewed By: ejguan Differential Revision: D33528259 Pulled By: IvanKobzarev fbshipit-source-id: b8c216558ca612bedd4c37205f38ed29c2c82b3c
lcy-seso · Jan 12, 2022 · 3c0c5bd · 3c0c5bd
1 parent e1f01d2
commit 3c0c5bd
Showing 1 changed file with 101 additions and 97 deletions.
diff --git a/binaries/CMakeLists.txt b/binaries/CMakeLists.txt
@@ -11,103 +11,107 @@ if(INTERN_BUILD_MOBILE)
   return()
 endif()
 
-#caffe2_binary_target("convert_caffe_image_db.cc")
-#caffe2_binary_target("convert_db.cc")
-#caffe2_binary_target("make_cifar_db.cc")
-#caffe2_binary_target("make_mnist_db.cc")
-#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("intra_inter_benchmark.cc")
-#target_include_directories(intra_inter_benchmark PUBLIC
-#  ${CMAKE_BINARY_DIR}/aten/src)
-#
-#caffe2_binary_target("at_launch_benchmark.cc")
-#target_include_directories(at_launch_benchmark PUBLIC
-#  ${CMAKE_BINARY_DIR}/aten/src)
-#
-#caffe2_binary_target("record_function_benchmark.cc")
-#target_include_directories(record_function_benchmark PUBLIC
-#  ${CMAKE_BINARY_DIR}/aten/src)
-#
-##caffe2_binary_target("predictor_verifier.cc")
-#caffe2_binary_target("print_registered_core_operators.cc")
-#caffe2_binary_target("run_plan.cc")
-#caffe2_binary_target("speed_benchmark.cc")
-#caffe2_binary_target("speed_benchmark_torch.cc")
-#caffe2_binary_target("compare_models_torch.cc")
-#caffe2_binary_target("split_db.cc")
-#
-#caffe2_binary_target("db_throughput.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})
-#  caffe2_binary_target("print_core_object_sizes_gpu.cc")
-#
-#  if(BUILD_TEST)
-#    # Core overhead benchmark
-#    caffe2_binary_target("core_overhead_benchmark_gpu.cc")
-#    target_link_libraries(core_overhead_benchmark_gpu benchmark ${CUDA_curand_LIBRARY})
-#  endif()
-#endif()
-#
-#if(USE_ROCM)
-#  caffe2_hip_binary_target("hip/inspect_gpu.cc")
-#  caffe2_hip_binary_target("hip/print_core_object_sizes_gpu.cc")
-#
-#  if(BUILD_TEST)
-#    # Core overhead benchmark
-#    caffe2_hip_binary_target("hip/core_overhead_benchmark_gpu.cc")
-#    target_link_libraries(core_overhead_benchmark_gpu benchmark)
-#  endif()
-#endif()
-#
-#if(USE_ZMQ)
-#  caffe2_binary_target("zmq_feeder.cc")
-#  target_link_libraries(zmq_feeder ${ZMQ_LIBRARIES})
-#endif()
-#
-#if(USE_MPI)
-#  caffe2_binary_target("run_plan_mpi.cc")
-#  target_link_libraries(run_plan_mpi ${MPI_CXX_LIBRARIES})
-#endif()
-#
-#if(USE_OPENCV AND USE_LEVELDB)
-#  caffe2_binary_target("convert_encoded_to_raw_leveldb.cc")
-#  target_link_libraries(
-#      convert_encoded_to_raw_leveldb
-#      ${OpenCV_LIBS} ${LevelDB_LIBRARIES} ${Snappy_LIBRARIES})
-#endif()
-#
-#if(USE_OPENCV)
-#  caffe2_binary_target("make_image_db.cc")
-#  target_link_libraries(make_image_db ${OpenCV_LIBS})
-#  caffe2_binary_target("convert_image_to_tensor.cc")
-#  target_link_libraries(convert_image_to_tensor ${OpenCV_LIBS})
-#endif()
-#
-#if(USE_OBSERVERS)
-#  caffe2_binary_target(caffe2_benchmark "caffe2_benchmark.cc" "benchmark_helper.cc")
-#endif()
-#
-#if(USE_OBSERVERS AND USE_OPENCV)
-#  caffe2_binary_target("convert_and_benchmark.cc")
-#  target_link_libraries(convert_and_benchmark ${OpenCV_LIBS})
-#endif()
-#
-## ---[ tutorials
-#caffe2_binary_target("tutorial_blob.cc")
-#
-#caffe2_binary_target("dump_operator_names.cc")
-#caffe2_binary_target("optimize_for_mobile.cc")
+if(BUILD_CAFFE2)
+  caffe2_binary_target("convert_caffe_image_db.cc")
+  caffe2_binary_target("convert_db.cc")
+  caffe2_binary_target("make_cifar_db.cc")
+  caffe2_binary_target("make_mnist_db.cc")
+  caffe2_binary_target("predictor_verifier.cc")
+  caffe2_binary_target("speed_benchmark.cc")
+  caffe2_binary_target("split_db.cc")
+  caffe2_binary_target("print_registered_core_operators.cc")
+
+  if(USE_OBSERVERS)
+    caffe2_binary_target(caffe2_benchmark "caffe2_benchmark.cc" "benchmark_helper.cc")
+    target_include_directories(caffe2_benchmark PUBLIC ${CMAKE_CURRENT_SOURCE_DIR}/../modules)
+  endif()
+
+  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)
+
+  caffe2_binary_target("run_plan.cc")
+  caffe2_binary_target("db_throughput.cc")
+
+  # ---[ tutorials
+  caffe2_binary_target("tutorial_blob.cc")
+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})
+  caffe2_binary_target("print_core_object_sizes_gpu.cc")
+
+  if(BUILD_TEST)
+    # Core overhead benchmark
+    caffe2_binary_target("core_overhead_benchmark_gpu.cc")
+    target_link_libraries(core_overhead_benchmark_gpu benchmark ${CUDA_curand_LIBRARY})
+  endif()
+endif()
+
+if(USE_ROCM)
+  caffe2_hip_binary_target("hip/inspect_gpu.cc")
+  caffe2_hip_binary_target("hip/print_core_object_sizes_gpu.cc")
+
+  if(BUILD_TEST)
+    # Core overhead benchmark
+    caffe2_hip_binary_target("hip/core_overhead_benchmark_gpu.cc")
+    target_link_libraries(core_overhead_benchmark_gpu benchmark)
+  endif()
+endif()
+
+if(USE_ZMQ)
+  caffe2_binary_target("zmq_feeder.cc")
+  target_link_libraries(zmq_feeder ${ZMQ_LIBRARIES})
+endif()
+
+if(USE_MPI)
+  caffe2_binary_target("run_plan_mpi.cc")
+  target_link_libraries(run_plan_mpi ${MPI_CXX_LIBRARIES})
+endif()
+
+if(USE_OPENCV AND USE_LEVELDB)
+  caffe2_binary_target("convert_encoded_to_raw_leveldb.cc")
+  target_link_libraries(
+      convert_encoded_to_raw_leveldb
+      ${OpenCV_LIBS} ${LevelDB_LIBRARIES} ${Snappy_LIBRARIES})
+endif()
+
+if(USE_OPENCV)
+  caffe2_binary_target("make_image_db.cc")
+  target_link_libraries(make_image_db ${OpenCV_LIBS})
+  caffe2_binary_target("convert_image_to_tensor.cc")
+  target_link_libraries(convert_image_to_tensor ${OpenCV_LIBS})
+endif()
+
+if(USE_OBSERVERS AND USE_OPENCV)
+  caffe2_binary_target("convert_and_benchmark.cc")
+  target_link_libraries(convert_and_benchmark ${OpenCV_LIBS})
+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)