- Enabled fc of oneDNN for bert test (#41235)

597d7efd · Jacek Czaja · GitHub · 705776ca · 597d7efd
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/inference/tests/api/analyzer_bert_tester.cc paddle/fluid/inference/tests/api/analyzer_bert_tester.cc +2 -0

未找到文件。
--- 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<std::vector<PaddleTensor>> outputs;