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

new -> zeros

Browse Source
This commit is contained in:
George Hotz
2020-11-09 17:31:42 -08:00
parent 16d564a53c
commit daf073535f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/opsgpu.py
View File

@@ -81,7 +81,7 @@ def cl_supsample_krnl_build(cl_ctx, result_op):
def supersample_op(ctx, input, out_shape, kernel_size, result_op):
(N, C, Yin, Xin), (Yout, Xout) = input.shape, out_shape[2:]
py,px = kernel_size
ret = buffer_new(ctx, out_shape)
ret = buffer_zeros(ctx, out_shape)
osize = np.array((Xout, Yout), dtype=cl.cltypes.uint2)
isize = np.array((Xin, Yin), dtype=cl.cltypes.uint2)
ksize = np.array((px, py), dtype=cl.cltypes.uint2)