未验证 提交 82b55961 编写于 作者: S Sylwester Fraczek 提交者: GitHub

add matmul_v2_transpose_reshape_fuse_pass to quant2_int8_mkldnn_pass.py (#37619)

上级 fab92824
...@@ -419,6 +419,7 @@ class Quant2Int8MkldnnPass(object): ...@@ -419,6 +419,7 @@ class Quant2Int8MkldnnPass(object):
if self._is_fc_quantized(graph): if self._is_fc_quantized(graph):
graph = self._apply_pass(graph, 'fc_mkldnn_pass') graph = self._apply_pass(graph, 'fc_mkldnn_pass')
graph = self._apply_pass(graph, 'matmul_transpose_reshape_fuse_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. # 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') graph = self._apply_pass(graph, 'runtime_context_cache_pass')
return graph return graph
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册