err, float32
MODIFY
tinygrad/ops.py
tinygrad/ops.py
@@ -96,7 +96,7 @@ class Ops:
ret = x.processing_op(op, y, C)
if 'LAZY' not in ctx.device: log_op(ProcessingOps, op, ret, [x, y])
Ops.flops += C.bs*C.cout*C.oy*C.ox*C.cin*C.H*C.W * 2
- Ops.mem += C.bs*C.cout*C.oy*C.ox + C.cout*C.cin*C.H*C.W + C.bs*C.cin*C.iy*C.ix
+ Ops.mem += (C.bs*C.cout*C.oy*C.ox + C.cout*C.cin*C.H*C.W + C.bs*C.cin*C.iy*C.ix) * 4
assert isinstance(ret, ctx.buffer)
assert ret.shape == C.out_shape
return ret
\ No newline at end of file

err, float32