7 Commits

Author	SHA1	Message	Date
nimlgen	31fca43706	kopt works with local+grouped reduce and tests (#1824)	2023-09-09 13:22:09 -07:00
George Hotz	ed194a1d3b	zero fold (#1748) ... * add constant fold * err, it's just zero folding * self store fold + caching * prints and more folds * simpler winograd kernels * remove childless uops	2023-09-03 13:48:11 -07:00
nimlgen	1c0449e190	add cache collector (#1595) ... * init cache collector * add test_cache_collector.py * switch GlobalCounters.cache to CacheCollector * init jit models test * jitted SD * add debug msg to print loaded bufs count * moved cache collctor to jit * clearer SD * no double device import	2023-08-28 19:59:55 -07:00
George Hotz	a6d842af7a	move device to ops (#1646) ... * move device to ops * mlops types * 2 lines	2023-08-23 08:30:17 -07:00
David Hou	4fbce972d7	CSE at uop level (#1483) ... * uop-level cse * add test * don't cache reduce alu ops * types * rename variable * fix * delete lines	2023-08-19 23:40:40 -07:00
David Hou	92754e177c	cache buffer loads across multiple bufs (#1482) ... * cache loads across buffers (since they may share rawbufs) * typing * add test * fix test * small changes to test * fix test * one big cache * whitespace * golf a line? * invalid is RawBuffer(0)[0], valid 1.	2023-08-19 09:09:58 -07:00
David Hou	56ee97b37f	dedup kernel args v2 (#1272) ... * new version * fix abstractions * try remove test * Revert "try remove test" This reverts commit 2fc18a9f8e. * assert_allclose * minimize the test * minimize the test * minimize the test * minimize the test * Revert "minimize the test" This reverts commit e0c0929596. * Revert "minimize the test" This reverts commit 88240551b1. * Revert "minimize the test" This reverts commit 78328a7ce2. * Revert "minimize the test" This reverts commit 989523fded. * skip test inside body * oops * oops	2023-07-18 20:03:42 -07:00