diff --git a/paddle/fluid/inference/tests/api/analyzer_dam_tester.cc b/paddle/fluid/inference/tests/api/analyzer_dam_tester.cc index 5ad6e4a8570b309e94375234d673e27698999cb7..fc87e0a8d1765f5a5c5bcfd21764b65165f35dc3 100644 --- a/paddle/fluid/inference/tests/api/analyzer_dam_tester.cc +++ b/paddle/fluid/inference/tests/api/analyzer_dam_tester.cc @@ -191,7 +191,9 @@ void profile(bool use_mkldnn = false) { if (use_mkldnn) { cfg.EnableMKLDNN(); - std::unordered_set op_list = {"conv3d"}; + // Enable all the mkldnn supported ops except conv3d in dam + std::unordered_set op_list = {"softmax", "elementwise_add", + "relu"}; cfg.SetMKLDNNOp(op_list); } @@ -235,7 +237,9 @@ void compare(bool use_mkldnn = false) { SetConfig(&cfg); if (use_mkldnn) { cfg.EnableMKLDNN(); - std::unordered_set op_list = {"conv3d"}; + // Enable all the mkldnn supported ops except conv3d in dam + std::unordered_set op_list = {"softmax", "elementwise_add", + "relu"}; cfg.SetMKLDNNOp(op_list); }