fix SRN algorithm infer error (#13851)

PaddlePaddle · Sep 11, 2024 · 1c8233d · 1c8233d
1 parent 4832bb6
commit 1c8233d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/infer/utility.py b/tools/infer/utility.py
@@ -291,6 +291,8 @@ def create_predictor(args, mode, logger):
         config.disable_glog_info()
         config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass")
         config.delete_pass("matmul_transpose_reshape_fuse_pass")
+        if mode == "rec" and args.rec_algorithm == "SRN":
+            config.delete_pass("gpu_cpu_map_matmul_v2_to_matmul_pass")
         if mode == "re":
             config.delete_pass("simplify_with_basic_ops_pass")
         if mode == "table":