add check

AlibabaPAI · Aug 26, 2024 · 0be6d23 · 0be6d23
1 parent bad3ee8
commit 0be6d23
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchacc/core/dynamic.py b/torchacc/core/dynamic.py
@@ -25,6 +25,8 @@ def mark_dynamic(x: torch.Tensor, dims: Union[List[int], int],
         bounds = [bounds]
     assert isinstance(dims, list), "dims should be of int or list type"
     assert isinstance(bounds, list), "bounds should be of int or list type"
+    assert len(dims) == len(
+        bounds), "dims and bounds should have the same length"
     for i, dim in enumerate(dims):
         if dim < (-x.dim()) or dim >= x.dim():
             raise ValueError(f"Dimension out of range (expected to be in range" \