add matmul_v2_transpose_reshape_fuse_pass to quant2_int8_mkldnn_pass.…

…py (PaddlePaddle#37619)
Zjq9409 · Dec 10, 2021 · bfec2e8 · bfec2e8
1 parent fea46a6
commit bfec2e8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/paddle/fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py b/python/paddle/fluid/contrib/slim/quantization/quant2_int8_mkldnn_pass.py
@@ -419,6 +419,7 @@ def _optimize_fp32_graph(self, graph):
         if self._is_fc_quantized(graph):
             graph = self._apply_pass(graph, 'fc_mkldnn_pass')
         graph = self._apply_pass(graph, 'matmul_transpose_reshape_fuse_pass')
+        graph = self._apply_pass(graph, 'matmul_v2_transpose_reshape_fuse_pass')
         # the following pass should be the last one since it will work on all fused ops.
         graph = self._apply_pass(graph, 'runtime_context_cache_pass')
         return graph