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

there is bug for inference using auto grwoth allocator (#21621)

test=develop
上级 73461a7a
...@@ -506,8 +506,6 @@ std::unique_ptr<PaddlePredictor> CreatePaddlePredictor< ...@@ -506,8 +506,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.
先完成此消息的编辑!
想要评论请 注册