Skip to content

Commit 3686572

Browse files
committed
fix SRN algorithm infer error
1 parent 33d5570 commit 3686572

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tools/infer/utility.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -291,6 +291,8 @@ def create_predictor(args, mode, logger):
291291
config.disable_glog_info()
292292
config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass")
293293
config.delete_pass("matmul_transpose_reshape_fuse_pass")
294+
if mode == "rec" and args.rec_algorithm == "SRN":
295+
config.delete_pass("gpu_cpu_map_matmul_v2_to_matmul_pass")
294296
if mode == "re":
295297
config.delete_pass("simplify_with_basic_ops_pass")
296298
if mode == "table":

0 commit comments

Comments
 (0)