diff --git a/tools/infer/utility.py b/tools/infer/utility.py index a93f364b31d6d9cb718337b0731330a172296396..c92e8e152a9ee4d86d269aec7ff5645f23cad443 100644 --- 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)