update the default cpu memory with MKLDNN

PaddlePaddle · Jun 19, 2018 · 9a25f28 · 9a25f28
1 parent f1f8327
commit 9a25f28
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/paddle/fluid/platform/cpu_info.cc b/paddle/fluid/platform/cpu_info.cc
@@ -30,7 +30,9 @@ DEFINE_double(fraction_of_cpu_memory_to_use, 1,
 
 DEFINE_uint64(initial_cpu_memory_in_mb,
 #ifdef PADDLE_WITH_MKLDNN
-              1000,
+              /* Aligned with mozga-intel, MKLDNN need at least 5000 MB
+               * to obtain the best performance*/
+              5000,
 #else
               500,
 #endif