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

oops

Browse Source
This commit is contained in:
George Hotz
2022-06-11 16:53:34 -07:00
parent 361123490a
commit c03c835d75
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/ops.py
View File

@@ -55,8 +55,9 @@ class Ops:
def reduce_op(ctx, op:BinaryOps, x, new_shape):
ret = ctx.buffer(new_shape)
ctx.op.reduce_op(op, x, ret)
log_op(op, ret, [x])
return ctx.op.reduce_op(op, x, ret)
return ret
def binary_op(ctx, op:ReduceOps, x, y):
ret = ctx.buffer(binary_broadcast(x.shape, y.shape))