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

assert and tuple

Browse Source
This commit is contained in:
George Hotz
2022-06-27 09:19:54 -07:00
parent 37a6c0ef59
commit dbf4aa09db
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
accel/lazy/ops_lazy.py
View File

@@ -72,6 +72,7 @@ class LazyBuffer:
# in lazy mode, we don't log until we realize
log_op(self.optype, [x.op for x in get_lazyops(self.op)], self.realized, real_srcs)
del self.op
assert self.realized.shape == self.shape
return self.realized
@staticmethod
@@ -156,7 +157,7 @@ def elementwise_op(op, srcs:Tuple[LazyBuffer]) -> LazyBuffer:
#return LazyBuffer(out_shape, BinaryOps, LazyOp(op, list(srcs)))
srcs = [srcs[0].op, srcs[1]]
#srcs = [srcs[0], srcs[1].op]
return LazyBuffer(out_shape, ProcessingOps, LazyOp(op, srcs))
return LazyBuffer(out_shape, ProcessingOps, LazyOp(op, tuple(srcs)))
if MERGE_ELEMENTWISE_OPS:
# remove the buffers from any BinaryOps that feed into this