148 commits

Author	SHA1	Message	Date
Christopher Milan	c8af163d2b	disable process replay by default (#16419) ... enable process replay with [pr] and assert with [PR] process replay no longer captures on master	2026-05-28 17:36:28 -04:00
nimlgen	e5891acab2	jit: precompile (#15848) ... * x * jit: precompile as sep step * x * s * x * x * x * ? * ? * x * x * viz * f * x * u * x * x	2026-04-23 00:23:32 +03:00
nimlgen	ae9b84d32f	rm beam uop (#15844)	2026-04-21 13:10:26 +03:00
qazal	12c653a743	remove opts arg in get_program, everything uses opts_to_apply [pr] (#15767) ... * check Ops.BEAM in process replay * remove opts from the get_program api * lint * simplify * cleanup	2026-04-16 22:42:43 +03:00
qazal	96092d110c	fix process_replay Ops.BEAM [pr] (#15752)	2026-04-16 07:35:28 +09:00
George Hotz	2611907afb	start ripping out old scheduler -- no maps (#14909) ... * start ripping out old scheduler -- no maps * no more metadata	2026-02-20 21:05:04 +08:00
qazal	9da7f5e733	disable process replay for AMD emulator renderer [pr] (#14766) ... * disable process replay for AMD emulator renderer [pr] * line * skip	2026-02-15 18:52:37 +09:00
George Hotz	c331798201	move tests to test/backend (#14691) ... * move tests to test/backend * fix imports * fix CI * revert that one * Fix formatting in README for test command	2026-02-12 11:09:44 +08:00
George Hotz	7a2a3b5c71	Remove Ops.KERNEL, it's all Ops.CALL now (#14603)	2026-02-07 10:21:54 +08:00
chenyu	b9fe8b7591	fix opt in process replay [pr] (#14599)	2026-02-06 16:49:56 -05:00
qazal	bdb421f13e	process_replay: passthrough sink arg for Ops.PROGRAM input (#14000)	2026-01-04 13:09:39 +09:00
qazal	366badaa68	require renderer argument in get_program, removes device opening in process replay [pr] (#13524)	2025-12-03 02:05:31 +08:00
George Hotz	18addc0a1d	process replay only get_program (#13475)	2025-11-27 08:18:18 -08:00
George Hotz	a8e005b095	enable process replay (non-checking) by default (#13474)	2025-11-27 07:28:44 -08:00
George Hotz	6fdbd03104	more divmod cleanup [p] (#13358) ... * more divmod cleanup [p] * lil cleanups, faster	2025-11-19 10:35:15 -08:00
George Hotz	385618d45b	skip process replay by default (#13353)	2025-11-19 08:25:34 -08:00
Sieds Lykles	e22c5e7e73	process_replay uses opts argument for KernelInfo.opts_to_apply (#12946) ... * opts_to_apply is opts * skip beamed kernels * simpler change * fix the tensor cores tests for process replay * use opts	2025-10-28 09:00:28 +01:00
qazal	fd51ecf983	process_replay for get_rangeify_map (#12624)	2025-10-12 15:14:40 +03:00
qazal	caae46cfba	fix process replay progress update (#12587)	2025-10-10 10:20:55 +03:00
qazal	ad49f8148b	switch process_replay to rangeify (#12509)	2025-10-08 11:26:43 +03:00
George Hotz	38dcadf07b	delete kernel.py (#12040) ... * delete kernel.py * delete that file * rip and tear * don't test search * imports * fix torch frontend * not a part of regen	2025-09-05 15:52:07 -07:00
George Hotz	12ab3f8b06	correct row_count in process replay (#11748)	2025-08-19 22:21:07 -07:00
qazal	16f0edbe90	pass opts arg in get_program process replay [pr] (#11571) ... * fix ptx process replay * keyword arg * renderer is also optional [pr] * test_linearizer fixup * name function order is args,ret,kwargs * can use opts_to_apply * pass through p.applied_opts * sink_arg * now it opens devices too	2025-08-08 03:05:09 +03:00
b1tg	06af9f9236	fix double exception + add name,loc in error msg (#11487) ... Co-authored-by: b1tg <b1tg@users.noreply.github.com>	2025-08-04 13:41:23 +03:00
George Hotz	842184a1ab	rename kernelize to schedule, try 2 (#11305)	2025-07-21 11:18:36 -07:00
qazal	3002c63b1e	process replay: optionally pass tinygrad import error (#11289) ... * process replay: optionally pass tinygrad import error * gate all tinygrad internals * s/getenv/os.getenv pre import * diff	2025-07-20 22:57:56 +03:00
qazal	ad155f5454	print inputs to get_program in process replay [pr] (#11051) ... * print inputs to get_program in process replay [pr] * colors * keep dataclass default escapes * Revert "keep dataclass default escapes" This reverts commit c6db7e8a7a. * note for ast_repr * add that back	2025-07-02 20:20:01 +03:00
qazal	452b22c9b6	fix process replay diff in PYTHON device [pr] (#11052) ... * fix process replay diff in PYTHON device [pr] The PYTHON backend pickles and encodes UOps, the encoded binary can't be directly diffed in process replay. * note	2025-07-02 11:06:46 +03:00
qazal	712980e167	fix extract_dataset + add tests to CI (#10995) ... * fix extract_dataset + tests * add CI * sops.gz itself is same as master * yml + gzip -c + ge * don't commit that * bump limit to 1000 * axis=7 * test_tiny	2025-06-27 01:51:36 +03:00
qazal	de4b9bf53b	add opts_to_apply option to AST KernelInfo (#10950) ... * proposal: add option to override opts in the get_program API * update test_linearizer_rewrite * state in uops * update process_replay and names * empty isn't none * fix process replay	2025-06-24 18:55:39 +03:00
qazal	7a5e4e0bf1	fix unittests process replay [pr] (#10947)	2025-06-24 10:30:23 +03:00
qazal	7820aeca8e	update codegen process replay to use get_program [pr] (#10921) ... * update codegen process replay to get_program [pr] * precommit * try str replace * +to_function_name * fixup tc * local2.sh * fix openpilot NOLOCALS * new local.sh * correct merge * beam cache * back * revert beam thing * adding opts_override and name_override makes output of get_program reproducible * min diff	2025-06-23 17:31:41 +03:00
George Hotz	b41e0563a3	move stuff to kernelize folder (#10902) ... * move stuff to kernelize folder * oops, forgot that	2025-06-20 16:10:20 -07:00
George Hotz	92678e59ee	move kernel to opt (#10899)	2025-06-20 15:22:28 -07:00
George Hotz	cba6e15937	split grouper and kernelize [pr] (#10854)	2025-06-17 17:54:20 -07:00
George Hotz	81b9c04574	move high level stuff to unit tests [pr] (#10708) ... * move high level stuff to unit tests [pr] * process replay on unit tests * fix pr, less compute * set omp num threads * set 200MB buffer size limit * delete junk * fix tests * faster * move test_indexing to unit * faster	2025-06-08 14:05:56 -07:00
qazal	910cabb081	add kernel count to grouper process replay differ [pr] (#10611)	2025-06-03 15:21:27 +03:00
qazal	3cc73a0172	simpler process replay main loop [pr] (#10588) ... * simpler process replay main loop [pr] * use logging * default to 1	2025-06-01 15:03:21 +03:00
qazal	dc882d3d7d	merge process replay and viz captures [pr] (#10581) ... * refactoring * test script * work * more work * diff * repr splits lines correctly * that * add location * add location * also don't need name_override * k.copy * [pr] * name_override 2 * err	2025-06-01 12:30:10 +03:00
George Hotz	0d39bb5de1	rename to get_kernelize_map (#10465)	2025-05-22 11:44:44 -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
qazal	9e2089dcd4	don't raise Exception in process replay [pr] (#10392) ... * don't raise Exception in process replay [pr] * continue generating diffs unless [pr] is set, exit(1) otherwise * change * works	2025-05-18 11:23:23 +03:00
qazal	e9e5b54e43	grouper cleanups and merge with insert_kernels [pr] (#10349) ... * grouper cleanups and merge with insert_kernels [pr] * remove that	2025-05-16 14:39:56 +03:00
qazal	1770e00c41	only CAPTURE_PROCESS_REPLAY=1 + add filterwarnings back [pr] (#10292)	2025-05-14 11:58:42 +03:00
qazal	1c97338be5	enable process replay assert for schedule [pr] (#10280) ... * enable process replay assert for schedule * start at unique+1	2025-05-14 11:10:47 +03:00
qazal	d13c100981	don't sort dims in verify_sink_dims [pr] (#10059) ... * don't sort dims in verify_sink_dims [pr] * 1 can exist with n * put process_replay warn last * assert shape is the same * bring that back	2025-04-26 23:24:30 +08:00
George Hotz	2ed3acd767	toposort is a function [pr] (#10004)	2025-04-23 16:25:03 +01:00
qazal	1d90be2cff	match kernelize API in process replay (#9948)	2025-04-21 05:23:41 +08:00
qazal	b58decac0c	fix diamond assigns before mapping tensors UOps to assigns (#9855) ... * keep tensor_map until diamond assign fixup * ctx	2025-04-18 14:17:43 +03:00
qazal	e201bc3e93	process replay kernel asts in toposort order [pr] (#9869) ... * process replay kernel asts in toposort order [pr] * use HEAD replay	2025-04-13 17:20:34 +08:00