diff --git a/tinygrad/codegen/linearizer.py b/tinygrad/codegen/linearizer.py index 6fd48e6f4a..71aaf44731 100644 --- a/tinygrad/codegen/linearizer.py +++ b/tinygrad/codegen/linearizer.py @@ -259,7 +259,7 @@ class Linearizer(Kernel): # load earlybufs loaded_buffers.update({b:self.global_load(self.bufs.index([lb for aliases in self.local_alias.values() for lb in aliases.values() - if lb is b][0]) if any(i in aliases for aliases in self.local_alias.values()) else i, + if lb is b][0]) if False else i, global_idxs+local_idxs+reduce_idxs+full_upcast_idxs) for i,b in enumerate(self.bufs) if b in self.earlybufs}) # run early AST (with reduce)