From 597d7efd0a3c32f0fd631bae8cdbef09fa4b0988 Mon Sep 17 00:00:00 2001 From: Jacek Czaja Date: Fri, 1 Apr 2022 08:32:26 +0200 Subject: [PATCH] - Enabled fc of oneDNN for bert test (#41235) --- paddle/fluid/inference/tests/api/analyzer_bert_tester.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddle/fluid/inference/tests/api/analyzer_bert_tester.cc b/paddle/fluid/inference/tests/api/analyzer_bert_tester.cc index 2570325c24a..8f7e5100922 100644 --- a/paddle/fluid/inference/tests/api/analyzer_bert_tester.cc +++ b/paddle/fluid/inference/tests/api/analyzer_bert_tester.cc @@ -156,6 +156,8 @@ void profile(bool use_mkldnn = false) { if (use_mkldnn) { config.EnableMKLDNN(); + config.pass_builder()->AppendPass("fc_mkldnn_pass"); + config.pass_builder()->AppendPass("fc_act_mkldnn_fuse_pass"); } std::vector> outputs; -- GitLab