mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

Reduce line count (#424)

Browse source 
* save a line, save a life

* save a line, save a life

* change order of tern
This commit is contained in:
Daniel Davis 2022-11-09 13:07:22 -05:00 • committed by GitHub
commit 64ff1ddc10
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tinygrad/mlops.py
View file

@ -64,8 +64,7 @@ class Max(Function):
max_is_1s = x.binary_op(BinaryOps.CMPEQ, ret.movement_op(MovementOps.EXPAND, x.shape))
# sum of locations, averaged
div = max_is_1s.reduce_op(ReduceOps.SUM, grad_output.shape)
div = div.movement_op(MovementOps.EXPAND, x.shape)
div = max_is_1s.reduce_op(ReduceOps.SUM, grad_output.shape).movement_op(MovementOps.EXPAND, x.shape)
max_is_amount = max_is_1s.binary_op(BinaryOps.DIV, div)
grad_output_expanded = grad_output.movement_op(MovementOps.EXPAND, x.shape)

3
tinygrad/tensor.py
View file

@ -19,8 +19,7 @@ class Tensor:
data = data.realize().toCPU()
if isinstance(data, np.ndarray):
if data.shape == tuple():
data = data.reshape((1,))
data = data if data.shape else data.reshape((1,))
self.lazydata = LazyBuffer.fromCPU(data.astype(np.float32), device)
elif isinstance(data, LazyBuffer):
self.lazydata = data