From c4238187be0f4d2e32e79f12aff5a4209dcb8cdb Mon Sep 17 00:00:00 2001 From: George Hotz Date: Fri, 10 Jun 2022 10:35:50 -0700 Subject: [PATCH] oops, we should have a linter in a precommit hook --- tinygrad/llops/ops_cpu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/llops/ops_cpu.py b/tinygrad/llops/ops_cpu.py index 8494416ce5..a3100ee6dd 100644 --- a/tinygrad/llops/ops_cpu.py +++ b/tinygrad/llops/ops_cpu.py @@ -100,7 +100,7 @@ def convdw(x,grad_output,dw,stride,groups): return dw def convdx(w,grad_output,dx,stride,groups): - C = get_conv_args(x.shape, dw.shape, stride, groups) + C = get_conv_args(dx.shape, w.shape, stride, groups) ggg = grad_output.reshape(C.bs, C.groups, C.rcout, C.oy, C.ox) tw = w.reshape(C.groups, C.rcout, C.cin, C.H, C.W) gdx = dx.reshape((C.bs, C.groups, C.cin, C.iy, C.ix))