From a908f447d55913d2f2665d503471281e8a247edf Mon Sep 17 00:00:00 2001 From: chenyu Date: Mon, 2 Feb 2026 08:34:45 -0500 Subject: [PATCH] remove disk special case in mstack_early_shrink [pr] (#14494) --- tinygrad/schedule/multi.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/tinygrad/schedule/multi.py b/tinygrad/schedule/multi.py index b9a41c3f77..5539967eb2 100644 --- a/tinygrad/schedule/multi.py +++ b/tinygrad/schedule/multi.py @@ -94,12 +94,7 @@ def mstack_early_shrink(ms:UOp, shrink:UOp): return s.shrink(tuple(new_arg)) for i, x in enumerate(ms.src): if x.op is Ops.COPY: - # if src device doesn't have a renderer, we have to view after the copy - # TODO: a way to understand this - if x.src[0].device in {"DISK", "NPY"}: - ret.append(apply_shrink(x, i)) - else: - ret.append(apply_shrink(x.src[0], i).copy_to_device(x.device)) + ret.append(apply_shrink(x.src[0], i).copy_to_device(x.device)) else: ret.append(apply_shrink(x, i).contiguous()) return ms.replace(src=tuple(ret))