From 09cc64eea70650bfaf58210f6861f84a1d65651f Mon Sep 17 00:00:00 2001 From: chenyu Date: Wed, 2 Jul 2025 16:36:40 -0400 Subject: [PATCH] remove const 0 clause in "UOp with size 0 is zero" [pr] (#11061) --- tinygrad/kernelize/kernelize.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tinygrad/kernelize/kernelize.py b/tinygrad/kernelize/kernelize.py index a1dfed5869..77b341cef2 100644 --- a/tinygrad/kernelize/kernelize.py +++ b/tinygrad/kernelize/kernelize.py @@ -51,8 +51,7 @@ def copy_reorder_view(copy:UOp, view:UOp, base:UOp): sym = symbolic_simple+PatternMatcher([ # UOp with size 0 is zero - (UPat(GroupOp.All-{Ops.SINK}, name="root"), lambda root: root.const_like(0) if root.base.st is not None and root.size == 0 \ - and not (root.base.op is Ops.CONST and root.base.arg == 0) else None), + (UPat(GroupOp.All-{Ops.SINK}, name="root"), lambda root: root.const_like(0) if root.base.st is not None and root.size == 0 else None), # DETACH and CONTIGUOUS_BACKWARD are NOOPs here (UPat((Ops.DETACH, Ops.CONTIGUOUS_BACKWARD), name="x"), lambda x: x.src[0]), # reduce of size 0 is the identity element