9 commits

Author	SHA1	Message	Date
chenyu	79b7799dba	clean up linearize schedule [pr] (#14565) ... * clean up linearize schedule [pr] don't mix ScheduleItem and UOp in schedule queue * ok	2026-02-05 15:24:09 -05:00
George Hotz	1afa3c0877	vmap on full model (#13340) ... * vmap on full model * vmap gemm * reduce sums on end * outer reduce * only if there's ranges * put those rules in symbolic * ranges * do opt later * add zero range	2025-11-18 16:06:06 -08:00
George Hotz	9c59b3d19e	vmap grad needs reduce_backward (#13336) ... * vmap grad needs reduce_backward * fuse and outer	2025-11-18 10:08:30 -08:00
George Hotz	06e39a88a9	outer vmap works (#13334) ... * outer vmap works * fuse works * vmap outer works * outer ranges work * grad work * should be good to merge	2025-11-18 09:27:48 -08:00
George Hotz	583560ab72	this is the right way to write vmap (#13328)	2025-11-17 20:20:52 -08:00
George Hotz	e4fead8a86	write scan in uops (#13321) ... * write scan in uops * ops range * no need for variable * meh, later * shorter	2025-11-17 16:58:08 -08:00
George Hotz	22c08b470c	fold using outerworld range (#13286) ... * scan using outerworld range * almost * sched * simple range * mypy * woooo outer range * spec passes * print the numbers * lol it runs * real test	2025-11-14 20:43:41 -08:00
George Hotz	7596c1b8f5	TestOuterworldReduce works (#12608)	2025-10-10 20:06:41 +08:00
George Hotz	5977df267f	outerworld uses expand (#12578)	2025-10-10 10:25:25 +08:00