fix v3 mkldnn

2022-04-27 07:48:14 +00:00 · 2022-04-27 07:48:14 +00:00 · f10b783e20
parent 882e6e549f
commit f10b783e20
1 changed files with 1 additions and 2 deletions
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@ -301,9 +301,8 @@ def create_predictor(args, mode, logger):
        # enable memory optim
        config.enable_memory_optim()
        config.disable_glog_info()
-        config.delete_pass("reshape_transpose_matmul_v2_mkldnn_fuse_pass")
-        config.delete_pass("reshape_transpose_matmul_mkldnn_fuse_pass")
        config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass")
+        config.delete_pass("matmul_transpose_reshape_fuse_pass")
        if mode == 'table':
            config.delete_pass("fc_fuse_pass")  # not supported for table
        config.switch_use_feed_fetch_ops(False)