Update the speed benchmark code

Summary: (for TIR demo cases) Closes facebookarchive/caffe2#1160 Differential Revision: D5761679 Pulled By: Yangqing fbshipit-source-id: 53b6c7fd098a394eba51baeac1e70371bcddf360
ezyang · Sep 2, 2017 · f2276d6 · f2276d6
1 parent 8b5fcfd
commit f2276d6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cmake/ProtoBuf.cmake b/cmake/ProtoBuf.cmake
@@ -26,6 +26,7 @@ function(custom_protobuf_find)
     message(STATUS "Using protobuf compiler ${PROTOBUF_PROTOC_EXECUTABLE}.")
   endif()
   set(Protobuf_FOUND TRUE PARENT_SCOPE)
+  set(Caffe2_IS_CUSTOM_PROTOBUF TRUE PARENT_SCOPE)
 endfunction()
 
 if (WIN32)