33 commits

Author	SHA1	Message	Date
George Hotz	92678e59ee	move kernel to opt (#10899)	2025-06-20 15:22:28 -07:00
George Hotz	411392dfb7	move files into uop dir (#10399) ... * move files into uop dir [pr] * tinygrad.uop is a thing * fix uop docs, no pr * fix viz	2025-05-18 11:38:28 -07:00
chenyu	6c30948df6	hand_coded_optimizations returns list[Opt] [pr] (#9938) ... new api looks like `k.apply_opts(hand_coded_optimizations(k))`	2025-04-19 20:26:59 -04:00
chenyu	8c6299bced	move hand_coded_optimizations to heuristic.py [pr] (#9844) ... * move hand_coded_optimizations to heuristic.py [pr] also folded all long lines * make a copy and rename self -> k * fix test	2025-04-10 23:40:16 -04:00
chenyu	f4f56d7c15	move time_linearizer to extra.optimization.helpers [pr] (#9048) ... no longer used in tinygrad	2025-02-12 15:49:58 -05:00
chenyu	a092b6395d	Tuple -> tuple, List -> list [pr] (#8936)	2025-02-06 14:21:19 -05:00
chenyu	b6be407bc6	fix handcode_opt bert [pr] (#8509) ... * fix handcode_opt bert [pr] * too slow	2025-01-05 19:14:12 -05:00
qazal	866dfa1f23	create_schedule([x.lazydata]) -> x.schedule() in tests (#8449)	2024-12-31 03:15:52 +08:00
George Hotz	8f95b578f6	use Estimates class [pr] (#8319) ... * use Estimates class [pr] * frozen dataclass	2024-12-18 10:19:32 -08:00
George Hotz	c8bf09b7d4	s/UOps/Ops (#7500) ... * s/UOps/Ops [pr] * fix	2024-11-03 11:26:10 +08:00
qazal	30989fb459	changes from the big graph branch [pr] (#7160) ... * metaops srcs * delete multioutput ctx var * always has metadata * shorter path for realized * this still needs inputs This reverts commit a59cbb2886.	2024-10-19 16:22:37 +03:00
George Hotz	a71bb09ec3	remove symbolic file [pr] (#7012)	2024-10-12 18:44:44 +08:00
chenyu	2fc26890c9	default BS=9 in handcode_opt bert (#6783) ... using 54 for 6 gpus now, and 2 is not a good default	2024-09-27 04:38:16 -04:00
wozeparrot	15cd42cfb9	feat: support TRACEMETA=2 in handcode_opt (#6767)	2024-09-26 16:58:29 +08:00
chenyu	0424c4967d	fix handcode_opt.py for bert (#6756)	2024-09-26 00:20:24 -04:00
qazal	935b6b658f	delete seen from the scheduler api [run_process_replay] (#6427) ... docs	2024-09-09 16:26:34 +08: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
George Hotz	8b34ee2f52	remove global_size and local_size from Kernel class [run_process_replay] (#5720) ... * remove global_size and local_size from Kernel class [run_process_replay] * sizes from the prg	2024-07-25 13:55:08 -07:00
George Hotz	7f5282b2f5	tests if the linearizer is generating dumb code (#5611) ... * tests if the linearizer is generating dumb code * push consts to the end * sort adds * sorted add and mul * this better * simple expand/contract * no math contract/expand	2024-07-20 20:36:32 -07:00
George Hotz	b399ccd6ef	BEAM bugfix, kernels dedup now (#5617) ... * BEAM bugfix, kernels dedup now * getenv is default	2024-07-20 19:43:50 -07:00
George Hotz	1113e47f96	print best in MCTS + light up the winner in hcopt	2024-07-20 09:39:36 -07:00
George Hotz	06e336bccb	mcts search (#5598) ... * mcts search * mcts cleanups * mcts cleanup * random shuffle children order * mcts in handcode_opt * src and remove_node * debug 3 to print ast * print the type * mcts in extra	2024-07-19 21:38:39 -07:00
George Hotz	2de82b8a5d	remove get_lazyop_info (#5570) ... * don't use get_lazyop_info more * keep that min * no ptx for that test	2024-07-19 03:05:33 -07:00
kormann	2c4add6844	pretty print lazy op per default (#5505) ... * pretty lop * min diff * walrus * fix * min diff * simplify * pretty helper function * ws * pretty uop upat * tests * stricter tests * test passes * ws * stronger upat test * delete print_tree * min diff * stricter exp test * fix merge * stronger uops eval test * +readable and deep upat test * +readable and deep upat test * sort inv fix * fix * revert allowed_len	2024-07-18 09:34:08 -07:00
George Hotz	fa7e734b49	MetaOps.KERNEL (#5543)	2024-07-17 19:41:23 -07:00
chenyu	4193095f67	fix handcode_opt.py with DEBUG=2 (#5530) ... only one ast per kernel now	2024-07-17 14:50:47 -04:00
George Hotz	03c2dc8bd7	lowerer is kernel [run_process_replay] (#5437)	2024-07-12 18:50:55 -07:00
chenyu	9a187e6102	fix handcode_opt script (#5435) ... * fix handcode_opt script * run in ci * real run in ci * HALF=0	2024-07-12 20:52:28 -04:00
George Hotz	870dc8c350	s/Linearizer/Lowerer [run_process_replay] (#5428)	2024-07-12 15:54:07 -07:00
George Hotz	6707c778d0	scheduleitem is not Tuple [run_process_replay] (#5425) ... * scheduleitem is not Tuple [run_process_replay] * fix tests * fix op + fuzzers * fix mop test	2024-07-12 15:13:19 -07:00
George Hotz	f6ef283e6a	s/loadops/metaops [run_process_replay] (#5421)	2024-07-12 13:26:50 -07:00
wozeparrot	d1cbd6bb95	unity handcode_resnet_opt and handcode_bert_opt (#5418)	2024-07-12 12:05:01 -07:00

Renamed from examples/handcode_bert_opt.py (Browse further)