8 Commits

Author	SHA1	Message	Date
qazal	982086f54c	UOps.VALID try 2 (#6623) ... * make UOps.VALID compile * fixable tests * bufs dedup * cleanup the CONST spec * regenerate dataset with graph_rewrite ```py def rewrite_const(const:UOp, st_src:UOp) -> UOp: st: ShapeTracker = st_src.arg return UOp(UOps.VALID, dtypes.bool, (st.to_uop(),)).where(UOp.const(const.dtype, const.arg), UOp.const(const.dtype, 0)) pm = PatternMatcher([(UPat(UOps.CONST, name="const", src=(UPat(UOps.SHAPETRACKER, name="st_src"),)), rewrite_const)]) ``` * rm arg * remove arg * revert arg removal This reverts commit 2c35c75c95. * red test_pickle_define_var	2024-09-21 14:19:25 +08:00
qazal	d2351af019	fixup non-void SINKs in tests [run_process_replay] (#6624)	2024-09-21 13:29:18 +08:00
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
George Hotz	dbd4536167	Revert "add UOps.VALID (#6387)" (#6441) ... This reverts commit 8186e4e7d6.	2024-09-09 21:33:00 +08:00
George Hotz	8186e4e7d6	add UOps.VALID (#6387) ... * uops valid * broke full_shape * fixup that st (hardcoded asts still red) * fixup DEFINE_VAR debug more debug * start moving stuff to ast_const * move test_linearizer * move test_linearizer_failures to ast_const * fixup test_schedule * small diff change * regenerate dataset * fixup test_multitensor * regen dataset try 2 --------- Co-authored-by: qazal <qazal.software@gmail.com>	2024-09-09 16:58:43 +08:00
qazal	f1bd2a5519	fix BUFFER_UOPS sts in verify_ast [run_process_replay] (#6389)	2024-09-06 16:59:22 +08:00
qazal	be6dda4093	hotfix: more lazyop rename to uop [run_process_replay] (#6157)	2024-08-18 17:28:44 +03:00
qazal	d9ce664350	add test_verify_ast [run_process_replay] (#6134)	2024-08-17 14:14:30 +03:00