failing test from application

test/test_linearizer_failures.py

@@ -1,11 +1,21 @@
 # ruff: noqa: E501
 import unittest
+from tinygrad import Tensor
 from tinygrad.uop.ops import UOp, Ops, AxisType
 from tinygrad.dtype import dtypes
 from tinygrad.engine.realize import get_program
 from tinygrad.device import Device
 
 class TestLinearizerFailures(unittest.TestCase):
+  def test_cumsum_repeat_reshape_multiply(self):
+    # cumsum + repeat + reshape + multiply fails when step > 512
+    step, num_steps = 513, 10
+    t = Tensor.arange(step).float()
+    phase = t.cumsum()
+    tiled = phase.repeat((num_steps,)).reshape(num_steps, step)
+    pattern = Tensor([1,0,0,1,0,0,0,0,1,0]).reshape(num_steps, 1)
+    result = (tiled * pattern).flatten()
+    result.numpy()  # should not raise AssertionError in CFGContext
   def test_fail_1(self):
     c0 = UOp(Ops.DEFINE_GLOBAL, dtypes.float.ptr(64), arg=0, src=())
     c1 = UOp.range(UOp.const(dtypes.index, 2), 1, AxisType.LOOP)