github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix type error

Browse Source
This commit is contained in:
George Hotz
2022-10-31 09:56:56 -07:00
parent 9585b6c0cf
commit c48fc47d01
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/lazy.py
View File

@@ -83,7 +83,7 @@ def _realize_binaryops(self:LazyBuffer) -> Tuple[DeviceBuffer, List[DeviceBuffer
if MERGE_ELEMENTWISE_INTO_REDUCE and src.realized is None and src.optype == BinaryOps and len(src.children) <= 1:
src = src.op
real_srcs[psrcs[0][0]] = LazyOp(psrcs[0][1].op.op, (src,), psrcs[0][1].op.arg)
for x in get_buffers(real_srcs[psrcs[0][0]]):
for x in get_buffers(real_srcs[psrcs[0][0]]): # type: ignore
# these are the early buffers
real_srcs[x] = x.realize(self.device)
op_type = ReduceOps