12 commits

Author	SHA1	Message	Date
qazal	a83710396c	support mselect input to CALL, less kernels in allreduce (#16567) ... * support mselect input to CALL, less kernels in allreduce * resolve mstack	2026-06-11 18:10:47 +09:00
qazal	21f1101691	add allreduce kernel count test (#16566)	2026-06-11 15:54:12 +09:00
qazal	fa400f9790	less E kernels in all2all (#16546)	2026-06-09 13:51:57 +09:00
qazal	b8931440ae	add all2all schedule test (#16545)	2026-06-09 12:41:35 +09:00
nimlgen	d3378010ee	schedule() -> schedule_linear() in tests (batch 1) (#15915) ... * schedule_with_vars -> linear_with_vars in tests * tests batch 1 * batch 2 * estimate_uop * simpler * rm	2026-04-24 23:40:53 +03:00
wozeparrot	2b8d303f75	allreduce in precast dtype (#15689)	2026-04-13 20:24:12 -07:00
George Hotz	baba274a76	minimal mstack pr to fix allreduce (#10649) ... * minimal mstack pr to fix allreduce * fix webgpu	2025-06-05 15:14:53 -07:00
George Hotz	411392dfb7	move files into uop dir (#10399) ... * move files into uop dir [pr] * tinygrad.uop is a thing * fix uop docs, no pr * fix viz	2025-05-18 11:38:28 -07:00
George Hotz	e13f2a3092	multi is O(1) (#10183) ... * multi is O(1) * allreduce * no new uops needed * junk * something * simple * that's really what i want * closer * inject _device_num * pretty print * cleanups * this * early dnum * ops allreduce is good * ish * device is the tuple and this is fine * simpler * progress * copy_multi * work * more tests * more tests pass * work * no None axis * tests * no none multi * type fixes * pre commit passes * lil * remove this * mlperf dataloader on mac * that test was wrong * unbind * support DEBUG=2 * realize * only unbind bound vars * don't include fixedvars * graph test * one test * fixedvars in hcq * new ring reduce * ring reduce * simpler ring * mselect * mselect doesn't work * Revert "mselect doesn't work" This reverts commit c78b77bd7d. * Revert "mselect" This reverts commit bb2e430ac3. * simpler * fixups * no optional * fix jit * move things around * cleanup multi * simpler multi * simpler reshape	2025-05-16 23:14:23 -07:00
George Hotz	876d2275a1	changes from new multi (#10353) ... * changes from new multi * revert hcq change	2025-05-16 13:07:29 -07:00
chenyu	e0ec8be37d	use CPU for test_schedule_ring (#9843) ... * use CPU for test_schedule_ring * why pre-commit is good	2025-04-10 23:20:53 -04:00
chenyu	d09897c2aa	allow double copy [pr] (#8559) ... fixed ring allreduce pattern and recovered most of the bert step time regression (10% faster), will double check all benchmark	2025-01-10 18:21:01 -05:00