332 commits

Author	SHA1	Message	Date
chenyu	eeee032b14	tiny cleanup of test_image_valid (#6597) ... * tiny cleanup of test_image_valid Sepcial and Variable to setup UOp * typo	2024-09-19 03:09:47 -04:00
George Hotz	012a2c449a	fix lt_folding VCONST issue [run_process_replay] (#6424) ... * le and ge [run_process_replay] * bugfix * fix divides bug * fix lt_folding issue	2024-09-19 14:59:20 +08:00
chenyu	496806ce75	another example of openpilot conv with valid (#6595)	2024-09-19 01:54:01 -04:00
chenyu	7f9fd556b0	_min_max for WHERE (#6564) ... prereq to gated load simplification just for int	2024-09-18 23:47:48 -04:00
chenyu	1b6eee02ad	failed test case for openpilot validhack conv (#6590) ... * failed test case for openpilot validhack conv can save 2ms once this is fixed * fix order	2024-09-18 23:12:30 -04:00
chenyu	bd40a26b8b	image valid test case that current approach does not work (#6584)	2024-09-18 06:06:03 -04:00
chenyu	162ead02a9	remove LOAD where valid is an empty set (#6579) ... 356 -> 354 valids	2024-09-18 03:49:41 -04:00
chenyu	a72d51e277	brute force VALIDHACK matching (#6575) ... * brute force VALIDHACK matching * cleanup * 9700	2024-09-18 01:59:50 -04:00
George Hotz	67a03e72bb	remove expr_idxs [run_process_replay] (#6567) ... * remove expr_idxs [run_process_replay] * goodbye that test	2024-09-17 18:34:51 +08:00
chenyu	b947db3de1	don't fold mul mod for common factor (#6566) ... it makes valid pattern more annoying	2024-09-17 06:01:27 -04:00
chenyu	5fb877c78c	generic valid match criteria of #6552 (#6558) ... 455 -> 364 valids. generalize `idx < image bound` to `idx < image bound + c` for some `c`	2024-09-17 02:40:36 -04:00
chenyu	c62b6fd8f0	match any statement in valid for simplification (#6554)	2024-09-17 01:39:47 -04:00
chenyu	7c942418a1	other side of simple out of bound valid case (#6552) ... 462 -> 455	2024-09-16 23:57:15 -04:00
chenyu	aeaf7894a7	more generic version of #6548 (#6549) ... x*(-1)<0 can be generalized to x*(-1)<c, 473 -> 462 valids	2024-09-16 23:17:16 -04:00
chenyu	596f41eb46	simple drop image valid case (#6548) ... * simple drop image valid case started unit test, 530 -> 473 valids * cleanup	2024-09-16 22:54:07 -04:00
George Hotz	07bd6e070d	add more uops tests for vmin/vmax/const_factor/divides (#6533)	2024-09-16 13:06:31 +08:00
George Hotz	21835fc08c	more graph rewrite tests (#6521)	2024-09-16 09:20:54 +08:00
George Hotz	cd90092f14	graph rewrite tests (#6519) ... * more graph rewrite tests * more complex test cases * more tests * more tests * cleanups * 9600 lines * cleanups	2024-09-15 17:29:16 +08:00
George Hotz	76487a3533	remove nop, use upat [run_process_replay] (#6489) ... * remove nop, use upat [run_process_replay] * mypy passes * no wonder nothing worked * fixes	2024-09-12 12:16:19 +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
chenyu	d9d1ae7248	more lt folding using gcd (#6469)	2024-09-11 02:09:35 -04:00
chenyu	15c4d4f406	fold unrolled arange div pattern (#6465)	2024-09-10 22:35:52 -04:00
chenyu	2105832b87	_min_max of MUL of 2 non-positive inputs (#6454)	2024-09-10 07:13:01 -04:00
chenyu	fcc69adfc5	simplify c0*x<c1 for negative int c0,c1 (#6431) ... * simplify c0*x<c1 for negative int c0,c1 * fine if rhs is zero	2024-09-09 21:05:53 -04:00
George Hotz	dbd4536167	Revert "add UOps.VALID (#6387)" (#6441) ... This reverts commit 8186e4e7d6.	2024-09-09 21:33:00 +08:00
George Hotz	42e5c8335e	remove args from min/max [run_process_replay] (#6430) ... * remove args from min/max [run_process_replay] * it's a ConstType * sconst_like unused * any const is fine	2024-09-09 18:18:20 +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
chenyu	ac98f5056e	move lt-folding to a function [run_process_replay] (#6422) ... and added more tests (some failed to match symbolic)	2024-09-09 02:04:52 -04:00
George Hotz	90fb17304f	put rewrite back in ops [run_process_replay] (#6421)	2024-09-09 13:53:51 +08:00
chenyu	25af78c593	failed uop_symbolic divmod test by variable (#6414)	2024-09-08 23:08:58 -04:00
chenyu	26c5d8346a	remove Variable from UOp.DEFINE_VAR (#6393) ... now it's just arg = (expr as str, min as UOp.const, max as UOp.const)	2024-09-06 05:55:19 -04:00
chenyu	9a9fea7b8c	move DEFINE_VAR min/max from src to arg (#6388) ... new arg is (Variable, min as CONST, max as CONST)	2024-09-06 05:01:02 -04:00
qazal	f1bd2a5519	fix BUFFER_UOPS sts in verify_ast [run_process_replay] (#6389)	2024-09-06 16:59:22 +08:00
chenyu	cc05016fa8	move test_pattern_matcher to test/unit (#6386)	2024-09-06 03:22:43 -04:00
chenyu	6fd24561d1	distribute MUL const into ADD for int (#6361) ... pre-req for real_stride	2024-09-05 01:36:57 -04:00
chenyu	a666450e4d	UOp pattern x + x -> x * 2 (#6224) ... * UOp pattern x + x -> x * 2 now there's no NEG, with this it covers all kinds of a*x+b*x * can remove x-x	2024-08-21 12:06:19 -04:00
chenyu	c9a9631818	no UnaryOps.NEG in generated UOp patterns (#6209) ... * no UnaryOps.NEG in generated UOp patterns removed pattern `x * (-1) -> -x` and `x != True` * those are fine because NEG became CMPNE and True * fix sd validation L2 norm	2024-08-21 11:08:22 -04:00
George Hotz	16f420f7a7	split full_graph_rewrite and linearize_uop [run_process_replay] (#6215) ... * split full_graph_rewrite and linearize_uop * fix tests * graph rewrite in test uops * add types	2024-08-20 20:12:33 -07:00
Max-We	53b20afa3f	Write tar_extract (#6180) ... * Add tar_extract * Add tar_extract tests * Fix dtype for initialization from path * Tests for path initialization * rm print --------- Co-authored-by: Maximilian Weichart <maximilian.weichart@icloud.com>	2024-08-19 12:06:17 -07:00
Eitan Turok	8556d0c642	Support gunzip in fetch (#6176) ... * init * update * clean * add type * clean * fix import order * shorten variable names	2024-08-19 12:04:40 -07:00
chenyu	00578a021b	re:6125 switch real_size to use uops [run_process_replay] (#6138) ... * switch real_size to use uops [run_process_replay] * enough to pass --------- Co-authored-by: George Hotz <geohot@gmail.com>	2024-08-19 13:20:24 -04:00
qazal	be6dda4093	hotfix: more lazyop rename to uop [run_process_replay] (#6157)	2024-08-18 17:28:44 +03:00
chenyu	f7950fc2b6	add E275 missing-whitespace-after-keyword linting rule (#6149) ... requires space after keywords like `assert`, `not`, `return`, `else`	2024-08-17 16:44:34 -04:00
George Hotz	88edc2902d	axis_is_masked with graph_rewrite [run_process_replay] (#6144)	2024-08-17 10:28:49 -07:00
qazal	d9ce664350	add test_verify_ast [run_process_replay] (#6134)	2024-08-17 14:14:30 +03:00
George Hotz	d9cb45af09	only axis is masked [run_process_replay] (#6123)	2024-08-16 21:01:17 -07:00
George Hotz	94aa5f11b5	Revert "use vmax for real_size [run_process_replay] (#6120)" (#6122) ... This reverts commit a6e3211444.	2024-08-16 20:33:19 -07:00
George Hotz	a6e3211444	use vmax for real_size [run_process_replay] (#6120) ... * use vmax for real_size [run_process_replay] * axis is masked	2024-08-16 20:17:23 -07:00
George Hotz	912f01ed4b	UOpGraph -> linearize_uop [run_process_replay] (#6119)	2024-08-16 19:48:39 -07:00
George Hotz	74ee9febec	remove iter from uopgraph (#6110) ... * remove iter from uopgraph * linearize returns uops * fix tests * linearize in linearize * tests fix * touchup * test failures	2024-08-16 15:58:29 -07:00