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))