From f10b783e2024cc2f302dedc2f1de0606b6e8489f Mon Sep 17 00:00:00 2001 From: tink2123 Date: Wed, 27 Apr 2022 07:48:14 +0000 Subject: [PATCH] fix v3 mkldnn --- tools/infer/utility.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index a93f364b..c92e8e15 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) -- GitLab