mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 00:15:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove unused Div op

Browse Source
This commit is contained in:
George Hotz
2020-12-06 13:02:12 -08:00
parent 20f95de408
commit c71a8ef222
1 changed files with 0 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tinygrad/ops.py
View File

@@ -45,19 +45,6 @@ class Mul(Function):
return unbroadcast(y*grad_output, x.shape), unbroadcast(x*grad_output, y.shape)
register('mul', Mul)
class Div(Function):
@staticmethod
def forward(ctx, x, y):
ctx.save_for_backward(x, y)
return x / y
@staticmethod
def backward(ctx, grad_output):
x,y = ctx.saved_tensors
return unbroadcast(grad_output / y, x.shape), unbroadcast(-x * grad_output / y**2, y.shape)
# TODO: registering this breaks the default div on the GPU
#register('div', Div)
class Pow(Function):
@staticmethod
def forward(ctx, x, y):