901 commits

Author	SHA1	Message	Date
George Hotz	d726eb6f48	uop resolve [run_process_replay] (#6826) ... * uop bool and int and stuff [run_process_replay] * add ne support * can't even be None anymore * BinaryOps.AND support * less compare	2024-10-01 13:11:42 +08:00
George Hotz	9dd9f71011	no global kernel stuff [run_process_replay] (#6808) ... * use traceback instead of global metadata crap [run_process_replay] * save the kernel * correct, imports clean, no device * UNPARENTED * speed * proudly unparented * Update ops.py * update tests for unparented --------- Co-authored-by: qazal <qazal.software@gmail.com>	2024-09-30 13:52:33 +08:00
qazal	12e4a4900a	hotfix: missing return in METAL dm benchmark (#6749)	2024-09-26 09:12:38 +08:00
qazal	b629a7998d	early assert buffer count limit [run_process_replay] (#6746) ... * better error message for buffer count limit [run_process_replay] * 3.9 needs that * assert ScheduleItem * new _test_buf_cnt	2024-09-26 08:24:26 +08:00
wozeparrot	c100f3d406	default threefry (#6116)	2024-09-25 17:45:13 +08:00
qazal	5ad2f95d01	process replay diff stats (#6736) ... * process replay diff stats * fix tuples	2024-09-25 15:19:56 +08:00
qazal	cefc3e9382	make all schedules immutable [run_process_replay] (#6718) ... * compute inputs and outputs in LBScheduleItem [run_process_replay] * simpler metadata, delete __hash__ * no dynamic field * test_diff_schedule	2024-09-24 21:08:16 +08:00
qazal	29330014ab	give FUZZ_SCHEDULE views a base (#6717) ... * memoryview to bytes * give FUZZ_SCHEDULE views a base	2024-09-24 19:20:37 +08:00
George Hotz	431ffc4254	hotfix: delete float16 failing	2024-09-23 17:42:57 +08:00
qazal	d24e4b1042	viz more kernel view work (#6659)	2024-09-23 10:48:35 +08:00
qazal	6be1bf09f1	hotfix: bring COMPARE_SCHEDULE=0 back (#6657)	2024-09-23 10:39:43 +08:00
qazal	6b65d8c461	more process replay tracing work [run_process_replay] (#6650)	2024-09-22 16:16:58 +08:00
qazal	5bafed2f88	process replay traceback (#6642)	2024-09-21 16:53:34 +08:00
chenyu	acef3e67fa	add an example that idx is const and valid cannot be removed (#6625) ... very weird	2024-09-20 05:46:27 -04:00
George Hotz	d4b662c318	new openpilot compile (#6573) ... * new openpilot compile * note, copyout doesn't work for images	2024-09-18 14:22:50 +08:00
chenyu	c3a70dbf0d	20 jitted steps in openpilot benchmark (#6577)	2024-09-18 02:15:16 -04:00
qazal	d8e5d5c663	move VIZ=1 tests to fuzzers (#6574)	2024-09-18 12:12:03 +08:00
George Hotz	28e565dc0d	prune independent kernels for openpilot [run_process_replay] (#6569) ... * prune independent kernels for openpilot [run_process_replay] * new pruning * prune first, then memory plan	2024-09-17 20:02:38 +08:00
qazal	455a27dd43	start viz unittests (#6550) ... * test_viz * more tests	2024-09-17 18:58:23 +08:00
qazal	9b9b83b8b0	viz tests (#6532) ... * vizz fuzz tests * caching * print timings * hotfix: update currentRewrite onClick * import from typing * indent into __main__	2024-09-16 13:08:42 +08:00
chenyu	1683b274b6	main example we want the valid removed (#6527) ... * main example we want the valid removed * ast lines are long	2024-09-15 21:49:10 -04:00
qazal	4ffb722d4e	var_vals prereq for deleting LBScheduleItem [run_process_replay] (#6511)	2024-09-14 17:00:30 +08:00
George Hotz	904f6a63fa	Revert "Revert "cleanup process_replay/* namings [run_process_replay] (#6429)…" (#6442) ... This reverts commit eda177da84.	2024-09-10 07:00:16 +08:00
George Hotz	eda177da84	Revert "cleanup process_replay/* namings [run_process_replay] (#6429)" (#6437) ... This reverts commit f4e83b30b4.	2024-09-09 18:52:36 +08:00
qazal	f4e83b30b4	cleanup process_replay/* namings [run_process_replay] (#6429)	2024-09-09 16:59:04 +08:00
qazal	935b6b658f	delete seen from the scheduler api [run_process_replay] (#6427) ... docs	2024-09-09 16:26:34 +08:00
qazal	9a67ec6174	refactor to list of kernels [run_process_replay] (#6403)	2024-09-08 17:19:45 +08:00
George Hotz	282af21b95	hotfix: DEBUG_EXPAND -1 and NOOPT in benchmark schedule	2024-09-06 17:22:30 +08:00
nimlgen	a1a15b54c9	qcom cache flush (#6367) ... * qcom cache flush * bench * linter * move	2024-09-05 13:23:39 +03:00
qazal	99018a4aa1	minor schedule differ utils [run_process_replay] (#6348) ... * minor schedule differ utils [run_process_replay] * rm	2024-09-04 03:41:38 +08:00
nimlgen	3adb76894d	validate image=2 float16=1 openpilot benchmark (#6346) ... * validate image=2 float=16 openpilot * linter * linter2	2024-09-03 20:13:40 +03:00
George Hotz	72939901fc	hotfix: ebs print kernel names	2024-08-29 21:20:36 -07:00
George Hotz	365babe391	precompute early_reject [run_process_replay] (#6327) ... * precompute early_reject [run_process_replay] * features for ebs * fix ocelot cache	2024-08-29 18:26:24 -07:00
qazal	dd4e5f1c8d	process replay rewrite (#6284) ... * process replay rewrite p2 * start some unittests + exceptions and exits * shebang * remove extra kernel init	2024-08-29 15:08:27 +03:00
qazal	ec34d9ee36	start benchmarking ast graph rewrite (#6297) ... * ast_rewrite to ctx var * add external_benchmark_ast * refactor to asts * track lazybuffers * more work * record checkpoint * cleanup	2024-08-27 18:18:44 +03:00
qazal	d2f8eeed2e	make [compare_schedule] the default [run_process_replay] (#6273) ... * make [compare_schedule] the default * capture ctx * logging * set capture to false	2024-08-26 21:40:03 +08:00
qazal	ae23540d6e	refresh process replay schedule ref in reset.py (#6265)	2024-08-24 16:12:51 +03:00
George Hotz	26498b322e	add BEAM to external_benchmark_schedule.py	2024-08-23 18:10:46 -07:00
chenyu	590c0922b6	Tensor.prod (#6250) ... * Tensor.prod a new reduce op! * onnx ReduceProd	2024-08-23 10:06:32 -04:00
chenyu	3fc8203475	remove NEG from handwritten ast in tests (#6234) ... * remove NEG from handwritten ast in tests * test_linearizer_failures	2024-08-22 09:06:59 -04:00
George Hotz	2c42e9c2c6	faster rewrite, no folder in expand/reduce [run_process_replay] (#6216) ... * faster rewrite, no folder in expand/reduce [run_process_replay] * is removing the expander there okay * parens * don't reconstruct exact match uop * fast do_reduce * expand pyint * most of the parents gains with less lines	2024-08-20 23:36:58 -07:00
qazal	074cf780dd	add option to only benchmark schedule [run_process_replay] (#6204)	2024-08-20 16:51:27 +03:00
qazal	478145cb8e	lowering error in diff_schedule is fine [run_process_replay] (#6185)	2024-08-19 20:51:12 +03:00
qazal	e28d29641f	more scheduler process replay tooling [run_process_replay] (#6178)	2024-08-19 15:35:51 +03:00
chenyu	b36a7273c6	RUF018 assignment-in-assert [run_process_replay] (#6172) ... assertion should not have side effect or `-O` breaks. initially just wanted to fix the one in rearrange, but it also made some long lines less long	2024-08-19 00:34:52 -04: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
qazal	28c75bf2a6	merge uops with ops (#6111) ... Co-authored-by: chenyu <chenyu@fastmail.com>	2024-08-16 18:17:57 -04:00
qazal	d5e3217076	hotfix: scheduler differ (#6115) ... * hotfix: scheduler differ * add the test back * track keys	2024-08-16 23:34:49 +03: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