minor cleanup

This commit is contained in:
George Hotz
2020-11-09 23:14:59 -08:00
parent d41ad2bf37
commit 55c914912d

View File

@@ -235,13 +235,9 @@ class Dot(Function):
__global const float *input,
__global const float *weight,
__global float *res,
int is0,
int is1,
int msize,
int ws0,
int ws1,
int osize
)
int is0, int is1, int msize,
int ws0, int ws1, int osize
)
{
int X = get_global_id(0); // isize
int Y = get_global_id(1); // osize
@@ -434,8 +430,6 @@ class MaxPool2D(Function):
raise NotImplementedError("GPU MaxPool2D.backward() not implemented")
register('max_pool2d', MaxPool2D, gpu=True)
# *** this is unfinished, fix this and TestMNIST.test_sgd_gpu should pass ***
class LogSoftmax(Function):
@staticmethod
def forward(ctx, input):