From 8b8c4df66ea03412a676067ae32ca7979f17023a Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 11 Aug 2026 14:51:12 -0700 Subject: [PATCH] weakint issue for symbolic --- tinygrad/schedule/rangeify2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/schedule/rangeify2.py b/tinygrad/schedule/rangeify2.py index 36be48e6d3..2582844fc8 100644 --- a/tinygrad/schedule/rangeify2.py +++ b/tinygrad/schedule/rangeify2.py @@ -154,7 +154,7 @@ class SplitCtx: range_number:int = -1 def _split_graph(ctx:SplitCtx, u:UOp) -> UOp: - assert len(u.shape) == 1, "rangeify needs to reduce to a single idx" + assert len(u.shape) <= 1, f"rangeify needs to reduce to a single idx, not {u.shape}" ctx.call_args.append(u) return u.param_like(len(ctx.call_args)-1)