Skip to content

Commit bf712b5

Browse files
authored
Merge pull request #5711 from hedaoyuan/inference
The pool_limit_size need be zero in mobile inference.
2 parents 6f22e6c + b34f21b commit bf712b5

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

paddle/math/Storage.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,13 @@ limitations under the License. */
1717
#include "paddle/utils/StringUtil.h"
1818
#include "paddle/utils/Util.h"
1919

20+
#ifndef PADDLE_MOBILE_INFERENCE
2021
DEFINE_int32(pool_limit_size,
2122
536870912,
2223
"maximum memory size managed by a memory pool, default is 512M");
24+
#else
25+
DEFINE_int32(pool_limit_size, 0, "default is 0");
26+
#endif
2327

2428
namespace paddle {
2529

0 commit comments

Comments
 (0)