The pool_limit_size need be zero in mobile inference.

7 years ago · b34f21bdf7
parent 6303d3ba3e
commit b34f21bdf7
1 changed files with 4 additions and 0 deletions
--- 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 {