Skip to content

Commit

Permalink
remove useless save
Browse files Browse the repository at this point in the history
  • Loading branch information
inisis committed Aug 16, 2024
1 parent ecd526e commit 9b697f5
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions onnxslim/third_party/symbolic_shape_infer.py
Original file line number Diff line number Diff line change
Expand Up @@ -3101,11 +3101,6 @@ def infer_shapes(in_mp, int_max=2**31 - 1, auto_merge=False, guess_output_rank=F
all_shapes_inferred = symbolic_shape_inference._infer_impl()
symbolic_shape_inference._update_output_from_vi()
if not all_shapes_inferred:
onnx.save_model(
symbolic_shape_inference.out_mp_,
"sym_shape_infer_temp.onnx",
save_as_external_data=True,
)
raise Exception("Incomplete symbolic shape inference")
return symbolic_shape_inference.out_mp_

Expand Down

0 comments on commit 9b697f5

Please sign in to comment.