Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "df167d87855f8b5e7bddc32ef3c3cd489027c7e1"
Commit 6e1cfb05 authored by MissPenguin's avatar MissPenguin
Browse files

fix cpu infer error for table structure

parent 3f92af45
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
tools/infer/utility.py
View file @ 6e1cfb05
......@@ -331,10 +331,11 @@ def create_predictor(args, mode, logger):
config.disable_glog_info()
config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass")
if mode == 'structure':
config.delete_pass("fc_fuse_pass") # not supported for table
config.switch_use_feed_fetch_ops(False)
config.switch_ir_optim(True)
if mode == 'structure':
config.switch_ir_optim(False)
# create predictor
predictor = inference.create_predictor(config)
input_names = predictor.get_input_names()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment