github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

test fix, wow fixing contig exposed so much

Browse Source
This commit is contained in:
George Hotz
2023-02-25 18:01:14 -08:00
parent c3d19101cb
commit 187738f64f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/lazy.py
View File

@@ -338,7 +338,8 @@ class LazyBuffer:
# now do the conv in this space and force it to be an image
ret = x.binary_op(BinaryOps.MUL, w).reduce_op(ReduceOps.SUM, (C.bs, C.oy, C.ox, C.cout//4, 4, 1, 1, 1, 1))
ret = ret.movement_op(MovementOps.RESHAPE, (C.bs*C.oy, C.ox*C.cout//4, 4)).contiguous()
ret = ret.movement_op(MovementOps.RESHAPE, (C.bs*C.oy, C.ox*C.cout//4, 4)) #.contiguous()
# NOTE: right now, this won't always be an image, otherwise the tests fail when we try to access it if it was SHRINKed
# undo hack for non multiples of 4 on C.rcout
if added_output_channels != 0: