tinygrad/test/test_multitensor.py at c4fdb9c725924fd1bc8a89ca07a1f405953b4d54

mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-11 07:05:04 -05:00

Files

chenyu 7948b05738 fix uneven shard with shrink and pad args on sharded axis (#5131 )

it's incorrect to assume all first (len(device)-1) shards would have the same size. e.g. size 2 shard 4 -> (1, 1, 0, 0)

2024-06-24 16:55:50 -04:00

View Raw