inductor: align inductor behavior with eager mode for split_with_sizes

ghstack-source-id: e9597778b26b2a16a62cdf8912bcbd95fbfa899a
Pull Request resolved: #99702

test/inductor/test_torchinductor.py

@@ -1863,6 +1863,14 @@ def fn(a, sizes):
         self.common(fn, (torch.randn(2, 2, 10), [4, 3, 3]))
         self.common(fn, (torch.randn(2, 2, 10), [1, 2, 3, 4]))
 
+    def test_split_with_sizes_failed(self):
+        @torch._dynamo.optimize("inductor")
+        def fn(a):
+            return torch.split(a, [2, 1, 1], dim=1)
+
+        with self.assertRaisesRegex(RuntimeError, ""):
+            fn(torch.randn(1, 5))
+
     def test_split(self):
         def fn(a):
             t = torch.split(a, 3, -1)

torch/_decomp/decompositions.py

@@ -1091,6 +1091,10 @@ def prod(x: List[int]):
 def split_with_sizes(
     self: Tensor, split_sizes: List[int], dim: int = 0
 ) -> List[Tensor]:
+    if sum(split_sizes) != self.shape[dim]:
+        raise ValueError(
+            "Split sizes don't add up to the tensor's size in the given dimension"
+        )
     num_splits = len(split_sizes)
     splits = []
     start_idx = 0