提交 a152a5c7 编写于 作者: B bingyanghuang 提交者: Tao Luo

Disable conv3d mkldnn in dam (#15335)

* disable conv3d mkldnn in dam

* Add some comments  test=develop
上级 73093656
...@@ -191,7 +191,9 @@ void profile(bool use_mkldnn = false) { ...@@ -191,7 +191,9 @@ void profile(bool use_mkldnn = false) {
if (use_mkldnn) { if (use_mkldnn) {
cfg.EnableMKLDNN(); 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); cfg.SetMKLDNNOp(op_list);
} }
...@@ -235,7 +237,9 @@ void compare(bool use_mkldnn = false) { ...@@ -235,7 +237,9 @@ void compare(bool use_mkldnn = false) {
SetConfig(&cfg); SetConfig(&cfg);
if (use_mkldnn) { if (use_mkldnn) {
cfg.EnableMKLDNN(); 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); cfg.SetMKLDNNOp(op_list);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册