5 commits

Author	SHA1	Message	Date
Roelof van Dijk	f88f71d73a	ruff: unnecessary-comprehension (#5174) ... * enable ruff C416 unnecessary-comprehension * already a list	2024-06-27 07:45:29 -04:00
David Hou	666a9c1448	don't view origin buffer when sharding (#5122) ... * make buffer view optional with a flag * do not view when sharding to save memory	2024-06-25 20:19:09 -07:00
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
chenyu	c0ba5e0dfb	multi copy_to_device return the copy on same device if possible (#5117) ... previously it always returns from the first device	2024-06-23 20:25:56 -04:00
George Hotz	5ba611787d	move image into tensor.py. delete features (#4603) ... * move image into tensor.py * change setup.py * openpilot tests need pythonpath now	2024-05-15 10:50:25 -07:00

Renamed from tinygrad/features/multi.py (Browse further)