Caffe2 flags needs to be used after the GlobalInit function is called

Summary: Pull Request resolved: pytorch#11120 Reviewed By: llyfacebook Differential Revision: D9598430 Pulled By: sf-wind fbshipit-source-id: 468f0ed7880339c9c4467d1cef29f5bc9fc80a2a
harouwu · Aug 31, 2018 · 26409a4 · 26409a4
1 parent a6cb414
commit 26409a4
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/binaries/benchmark_helper.cc b/binaries/benchmark_helper.cc
@@ -329,7 +329,6 @@ int benchmark(
     bool FLAGS_text_output,
     int FLAGS_warmup,
     bool FLAGS_wipe_cache) {
-  caffe2::GlobalInit(&argc, &argv);
   // Check arguments to be correct
   {
     // Need to check whether file exists, as the file reader does not assert if

diff --git a/binaries/caffe2_benchmark.cc b/binaries/caffe2_benchmark.cc
@@ -77,6 +77,7 @@ CAFFE2_DEFINE_bool(
     "Whether to evict the cache before running network.");
 
 int main(int argc, char** argv) {
+  caffe2::GlobalInit(&argc, &argv);
   benchmark(
       argc,
       argv,