Skip to content

Commit 91e8fba

Browse files
committed
Enable MKLDNN in Resnet50Tester
test=develop
1 parent 582f59c commit 91e8fba

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

paddle/fluid/inference/tests/api/analyzer_resnet50_tester.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,9 @@ void SetConfig(AnalysisConfig *cfg) {
2727
cfg->device = 0;
2828
cfg->enable_ir_optim = true;
2929
cfg->specify_input_name = true;
30+
#ifdef PADDLE_WITH_MKLDNN
31+
cfg->_use_mkldnn = true;
32+
#endif
3033
}
3134

3235
void SetInput(std::vector<std::vector<PaddleTensor>> *inputs) {

0 commit comments

Comments
 (0)