Disable conv3d mkldnn in dam (#15335)

* disable conv3d mkldnn in dam * Add some comments test=develop
6 years ago · a152a5c731
parent 73093656b8
commit a152a5c731
1 changed files with 6 additions and 2 deletions
--- 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<std::string> op_list = {"conv3d"};
+    // Enable all the mkldnn supported ops except conv3d in dam
+    std::unordered_set<std::string> 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<std::string> op_list = {"conv3d"};
+    // Enable all the mkldnn supported ops except conv3d in dam
+    std::unordered_set<std::string> op_list = {"softmax", "elementwise_add",
+                                               "relu"};
    cfg.SetMKLDNNOp(op_list);
  }