Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 91e8fbac 编写于 作者: M Michal Gallus
浏览文件
操作

Enable MKLDNN in Resnet50Tester 

test=develop
上级 582f59c1
显示空白变更内容
内联 并排
Showing with 3 addition and 0 deletion
paddle/fluid/inference/tests/api/analyzer_resnet50_tester.cc
浏览文件 @ 91e8fbac
...@@ -27,6 +27,9 @@ void SetConfig(AnalysisConfig *cfg) { ...@@ -27,6 +27,9 @@ void SetConfig(AnalysisConfig *cfg) {
cfg->device = 0; cfg->device = 0;
cfg->enable_ir_optim = true; cfg->enable_ir_optim = true;
cfg->specify_input_name = true; cfg->specify_input_name = true;
#ifdef PADDLE_WITH_MKLDNN
cfg->_use_mkldnn = true;
#endif
} }
void SetInput(std::vector<std::vector<PaddleTensor>> *inputs) { void SetInput(std::vector<std::vector<PaddleTensor>> *inputs) {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或