update the default cpu memory with MKLDNN

7 years ago · 9a25f2895c
parent f1f8327c31
commit 9a25f2895c
1 changed files with 3 additions and 1 deletions
--- 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