github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-26 15:28:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

concerning, does this pass

Browse Source
This commit is contained in:
qazal
2024-05-28 23:56:07 +03:00
parent 02bdb650e6
commit 64d4664f17
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/codegen/linearizer.py
View File

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