Skip to content

Commit

Permalink
[alt] fix for issue with input generation (nod-ai#361)
Browse files Browse the repository at this point in the history
In
<nod-ai@776822a>,
a line of code was erroneously removed, resulting in a failure to get
the input shape for static dims.

This adds back in the missing line to append the static shapes to the
int_dims list.
  • Loading branch information
zjgarvey authored Oct 3, 2024
1 parent 776822a commit 11158e6
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions alt_e2eshark/e2e_testing/onnx_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ def get_node_shape_from_dim_param_dict(node: onnxruntime.capi.onnxruntime_pybind
raise ValueError(
f"input node '{node.name}' has a non-positive dim: {dim}. Consider setting cutsom inputs for this test."
)
int_dims.append(dim)
return int_dims


Expand Down

0 comments on commit 11158e6

Please sign in to comment.