From a152a5c73140130eaa1cd037a92014f504acc558 Mon Sep 17 00:00:00 2001 From: bingyanghuang <33643817+bingyanghuang@users.noreply.github.com> Date: Tue, 15 Jan 2019 20:49:11 +0800 Subject: [PATCH] Disable conv3d mkldnn in dam (#15335) * disable conv3d mkldnn in dam * Add some comments test=develop --- paddle/fluid/inference/tests/api/analyzer_dam_tester.cc | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/inference/tests/api/analyzer_dam_tester.cc b/paddle/fluid/inference/tests/api/analyzer_dam_tester.cc index 5ad6e4a85..fc87e0a8d 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); } -- GitLab