未验证 提交 d0943dbe 编写于 作者: Z Zhaolong Xing 提交者: GitHub

CHERRY_PICK: Fix the bug for inference when using auto grwoth allocator (#21623)

test=release/1.6
上级 1f598dfa
...@@ -505,8 +505,6 @@ std::unique_ptr<PaddlePredictor> CreatePaddlePredictor< ...@@ -505,8 +505,6 @@ std::unique_ptr<PaddlePredictor> CreatePaddlePredictor<
std::string flag = "--fraction_of_gpu_memory_to_use=" + std::string flag = "--fraction_of_gpu_memory_to_use=" +
std::to_string(fraction_of_gpu_memory); std::to_string(fraction_of_gpu_memory);
flags.push_back(flag); flags.push_back(flag);
// use auto growth strategy here.
flags.push_back("--allocator_strategy=auto_growth");
flags.push_back("--cudnn_deterministic=True"); flags.push_back("--cudnn_deterministic=True");
VLOG(3) << "set flag: " << flag; VLOG(3) << "set flag: " << flag;
framework::InitGflags(flags); framework::InitGflags(flags);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册