Add check for input rank

Signed-off-by: gcunhase <4861122+gcunhase@users.noreply.github.com>

Author: gcunhase

modelopt/onnx/autocast/referencerunner.py

@@ -92,9 +92,13 @@ def _validate_inputs(self, data_loader):
                 # Get model and data shapes as numpy arrays
                 inp_shape_model = np.array(self.input_shapes[inp_name])
                 inp_shape_data = np.array(inp_shape.shape)
-                # Skip check for unknown dimensions (shape = -1)
-                mask = (inp_shape_model != -1) & (inp_shape_data != -1)
-                if np.any(inp_shape_model[mask] != inp_shape_data[mask]):
+                # Compare input rank
+                raise_value_error = len(inp_shape_model) != len(inp_shape_data)
+                if not raise_value_error:
+                    # Compare input shape, skipping check for unknown dimensions (shape = -1)
+                    mask = (inp_shape_model != -1) & (inp_shape_data != -1)
+                    raise_value_error = np.any(inp_shape_model[mask] != inp_shape_data[mask])
+                if raise_value_error:
                     raise ValueError(
                         f"Input shape from '{inp_name}' does not match provided input shape: "
                         f"{self.input_shapes[inp_name]} vs {list(inp_shape.shape)}. "