weakint issue for symbolic

2026-08-11 14:51:12 -07:00
parent cddd0f8083
commit 8b8c4df66e
1 changed files with 1 additions and 1 deletions
@@ -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)