Skip to content

Commit 8c26e5b

Browse files
committed
Update: consider input shape > 0 as static
Signed-off-by: gcunhase <4861122+gcunhase@users.noreply.github.com>
1 parent 629e4bb commit 8c26e5b

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

modelopt/onnx/autocast/referencerunner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,8 +95,8 @@ def _validate_inputs(self, data_loader):
9595
# Compare input rank
9696
raise_value_error = len(inp_shape_model) != len(inp_shape_data)
9797
if not raise_value_error:
98-
# Compare input shape, skipping check for unknown dimensions (shape = -1)
99-
mask = inp_shape_model != -1
98+
# Compare input shape, skipping check for unknown dimensions
99+
mask = inp_shape_model > 0
100100
raise_value_error = np.any(inp_shape_model[mask] != inp_shape_data[mask])
101101
if raise_value_error:
102102
raise ValueError(

0 commit comments

Comments
 (0)