From b34f21bdf787a01aa271af260dadacbcda0165a6 Mon Sep 17 00:00:00 2001 From: hedaoyuan Date: Thu, 16 Nov 2017 19:25:51 +0800 Subject: [PATCH] The pool_limit_size need be zero in mobile inference. --- paddle/math/Storage.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/paddle/math/Storage.cpp b/paddle/math/Storage.cpp index 4adaaef983..a2ef731ecb 100644 --- a/paddle/math/Storage.cpp +++ b/paddle/math/Storage.cpp @@ -17,9 +17,13 @@ limitations under the License. */ #include "paddle/utils/StringUtil.h" #include "paddle/utils/Util.h" +#ifndef PADDLE_MOBILE_INFERENCE DEFINE_int32(pool_limit_size, 536870912, "maximum memory size managed by a memory pool, default is 512M"); +#else +DEFINE_int32(pool_limit_size, 0, "default is 0"); +#endif namespace paddle { -- GitLab