7 Commits

Author	SHA1	Message	Date
George Hotz	bdd0c06f29	add void type to uop (#6471) ... * unwrap_dtype maybe * uopgraph stuff that hardcoded None * test_ops passes * dtypes.py fixups * update test_linearizer and friends * more ast updates * test_beam and test_schedule too * add void type to uop [run_process_replay] * remove dumb casts * start making it green * more cast cleanups * more cls methods to fix * regenerate dataset * split UOp and NOp const * maybe that too * fix docs * update test_uop_symbolic * test_verify_ast * new sops with no diff * meh, type_ignore is alright * remove that assert --------- Co-authored-by: qazal <qazal.software@gmail.com>	2024-09-11 18:16:28 +08:00
qazal	bcb2f1caa3	init REDUCE_AXIS with BinaryOps (#6256) ... * REDUCE_AXIS arg with BinaryOps * more work in kernel.py fixup sops.gz * fix TestGraphRewriteEfficiency	2024-08-24 11:28:41 +03:00
qazal	0d4887e9df	use UOps.WMMA everywhere (#6255) ... * add UOps.WMMA_AXIS * delete ReduceOps.WMMA from ops	2024-08-23 15:03:26 -04:00
George Hotz	9bc81c6db4	UOps.SHAPETRACKER (#6129) ... * UOps.SHAPETRACKER [run_process_replay] * no process replay	2024-08-16 23:26:34 -07:00
George Hotz	89c7989659	no shapetracker in ops [run_process_replay] (#6117)	2024-08-16 17:23:27 -07:00
qazal	28c75bf2a6	merge uops with ops (#6111) ... Co-authored-by: chenyu <chenyu@fastmail.com>	2024-08-16 18:17:57 -04:00
qazal	c23d44c779	AST is UOp (#6030) ... * most of the work from the uops2 branch * schedule * realize * kernel * lowerer * search * green * merge uops with ops * Revert "merge uops with ops" This reverts commit 1408a59f12. * fix benchmark * remove extra dedup	2024-08-16 22:09:00 +03:00