Skip to content

Commit 9a25f28

Browse files
committed
update the default cpu memory with MKLDNN
1 parent f1f8327 commit 9a25f28

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

paddle/fluid/platform/cpu_info.cc

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,9 @@ DEFINE_double(fraction_of_cpu_memory_to_use, 1,
3030

3131
DEFINE_uint64(initial_cpu_memory_in_mb,
3232
#ifdef PADDLE_WITH_MKLDNN
33-
1000,
33+
/* Aligned with mozga-intel, MKLDNN need at least 5000 MB
34+
* to obtain the best performance*/
35+
5000,
3436
#else
3537
500,
3638
#endif

0 commit comments

Comments
 (0)