70 commits

Author	SHA1	Message	Date
George Hotz	b4eb876d5a	kernel.py no longer permutes reduce axis [pr] (#10968) ... * kernel.py no longer permutes reduce axis [pr] * delete tests that handcode uops * regen of sops is broken... * put import back * just remove that * disable those tests	2025-06-26 17:44:58 -07:00
George Hotz	92678e59ee	move kernel to opt (#10899)	2025-06-20 15:22:28 -07:00
chenyu	652db5702b	move test_conv_shapetracker and some test_search util into unit test (#10812)	2025-06-14 13:29:32 -04:00
George Hotz	32e9949052	rename lazydata to uop (#10698)	2025-06-08 08:42:22 -07:00
qazal	5b59728c75	refactor LOAD(DEFINE_GLOBAL, VIEW) in kernels to LOAD(VIEW(DEFINE_GLOBAL)) (#10541) ... * changes to core tinygrad * fixups pt1 TC=3 docs/abstractions2.py IMAGE=2 test_quantize_dsp test_schedule * more tests * green now * images stay images	2025-05-30 14:27:58 +03:00
qazal	7051bf3fd5	fixup hardcoded asts ptr dtype and constants [pr] (#10570) ... * fixup hardcoded asts ptr dtype and constants [pr] * use kernelize for test_kernel_count	2025-05-30 09:38:32 +03: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	01e8b60911	acc_dtype -> dtype (#9402) ... matched numpy and torch	2025-03-10 16:05:30 -04:00
qazal	2eab8021fb	remove inputs+outputs attributes from ScheduleItem [pr] (#9192) ... * remove inputs/outputs from ScheduleItem * fix test_linearizer * fix test_conv_shapetracker * fix test_schedule + lint * test_image_dtype + multitensor + search	2025-02-21 13:48:11 +01: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
Ahmed Harmouche	133cacadde	Autogen webgpu dawn, removing wgpu-py dependency (f16 support part 1) (#8646) ... * Switch to dawn, all tests passing locally * Use dawn-python * Skip failing test * Skip midcast and fix timestamp on metal ci * Autogen webgpu * Try fetch dawn lib again * /usr/lib * Without lib prefix * Test autogen diff * Delete webgpu support, move everything to ops_webgpu * mypy fix * Simplify, refactor * Line savings * No ResultContainer * Type annotation for result * Some more simplifications * Why was this explicit sync used at all? * Refactor: delete functions that are only used once * Create shader module inline * Clear unit tests cache, maybe that solves it * That wasn't it * Try deleting cache to pass failing weight compare * weights_only=False for pytorch 2.6 * Simplify ctype array creation * Remove nanosecond precision timestamps * Simplify error handling * Refactor, add back type annotations * Deleted custom submit function, refactor * read_buffer simplify * Fix use after free, refactor * Simplify supported_features * Runtime docs --------- Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>	2025-02-07 15:16:59 +08:00
Ignacio Sica	aec3b8d515	add regression test: test_get_kernel_actions_preserves_actions_state (#8907) ... * test_get_kernel_actions_preserves_actions_state * simplify * simplify * refactor assert message	2025-02-05 14:13:01 -05:00
Ignacio Sica	15f94ac964	TC_SEARCH_OVER_SHAPE to search multiple TC shapes (#8793) ... * squash search over search * refactor assert * init benchmark * cleaner get_kernel_actions * cleaner get_kernel_actions * add comment	2025-02-05 11:03:46 -05:00
Ignacio Sica	b240f12593	[TIP-9] rename Opt's amt to arg 2 (#8770) ... * rename Opt amt to arg * ignore_beam_cache for test_tiny * move ignore_beam_cache to test_tiny * move to separate pr * revert space change --------- Co-authored-by: chenyu <chenyu@fastmail.com>	2025-01-27 14:19:04 -05:00
George Hotz	3ed146a5ff	Revert "rename Opt amt to arg (#8767)" (#8769) ... This reverts commit bf041659a5.	2025-01-27 23:46:37 +09:00
Ignacio Sica	bf041659a5	rename Opt amt to arg (#8767)	2025-01-27 23:36:47 +09:00
qazal	866dfa1f23	create_schedule([x.lazydata]) -> x.schedule() in tests (#8449)	2024-12-31 03:15:52 +08:00
George Hotz	275951b730	clean up a few parents -> toposort [pr] (#7984) ... * clean up a few parents -> toposort [pr] * rename to old_parents + sched tests * a few more * that one * second to last * final	2024-12-02 15:59:31 +08:00
ignaciosica	597a239e28	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725) ... * remove unaryops * remove ternaryops * remove metaops * hotfix * remove binaryops * hotfix: test_pattern_matcher --------- Co-authored-by: qazal <77887910+Qazalin@users.noreply.github.com>	2024-11-16 20:56:56 +08:00
Carl Basho	630a7f37cf	update tests (#7554) ... Co-authored-by: John Doe <null@mail.com> Co-authored-by: chenyu <chenyu@fastmail.com>	2024-11-05 11:35:15 -05:00
George Hotz	c8bf09b7d4	s/UOps/Ops (#7500) ... * s/UOps/Ops [pr] * fix	2024-11-03 11:26:10 +08:00
George Hotz	d9d4dd6756	faster ci [pr] (#7348)	2024-10-29 14:01:44 +08:00
George Hotz	ded1b38b84	minor dtype cleanup [pr] (#7124) ... * minor dtype cleanup [pr] * use ptr() function	2024-10-17 17:41:23 +08:00
qazal	20d3c2d113	unify UOps.SHAPETRACKER and UOps.SWIZZLE with UOps.VIEW (#6955) ... * add UOps.VIEW * update hardcoded asts * update sops.gz	2024-10-09 02:00:17 +08:00
wozeparrot	c100f3d406	default threefry (#6116)	2024-09-25 17:45:13 +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
qazal	442150a8df	more ast_const for hardcoding consts [run_process_replay] (#6418)	2024-09-09 11:35:08 +08:00
gswangg	1dc6040877	migrate test_search.py to UOp AST (#6245) ... * add imports and update test_kernel_count with UOp AST * test_filter_global_buffer * remove LazyOp * remove extra.ops and ReduceOps --------- Co-authored-by: qazal <77887910+Qazalin@users.noreply.github.com>	2024-08-24 16:13:53 +03:00
qazal	d1d41130cd	use membufs in ImageDType checks [run_process_replay] (#6136) ... * use membufs in ImageDType checks * set by key [run_process_replay]	2024-08-17 16:17:46 +03: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
qazal	4d38fec8c1	rename lazyops to parents [run_process_replay] (#6091)	2024-08-15 17:27:32 +03: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	fa7e734b49	MetaOps.KERNEL (#5543)	2024-07-17 19:41:23 -07:00
chenyu	28972418c4	s/get_linearizer/get_kernel [run_process_replay] (#5467)	2024-07-13 20:32:22 -04:00
George Hotz	03c2dc8bd7	lowerer is kernel [run_process_replay] (#5437)	2024-07-12 18:50:55 -07: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
qazal	28bf8d86d8	test_linearizer with multi output ASTs (#5115) ... * ast is tuple * run test_phi_simplification * update reason * more tc * beam * a few more * use test_opt directly	2024-06-23 15:41:24 +03:00
George Hotz	9f875123b6	small changes from lowerer. [run_process_replay] [no_assert] (#5102)	2024-06-22 11:09:35 -07:00
nimlgen	fd071ba27e	amd mockgpu correct timer resolution (#4942) ... * amd mockgpu correct timer resolution * test it	2024-06-13 10:07:34 +03:00
qazal	8b5bcf309a	process replay in all of CI (#4884)	2024-06-10 14:49:29 -04:00
qazal	f64fa51a64	process replay for test/* (#4799) ... * add input to unit tests [run_process_replay] * add setup [run_process_replay] * run tests [run_process_replay] * add cuda and amd [run_process_replay] * run everything but BEAM=2 [run_process_replay] * skip export_model [run_process_replay] * fix amd CI * add concurrency back	2024-06-03 12:01:58 +03:00
chenyu	456aa0b656	update test_search kernel count (#4652) ... integration test that beaming 1 kernel increments kernel count by 1, and moved exiting test_kernel_count to TestTimeLinearizer	2024-05-19 13:54:52 -04:00
Léo	967e35f8b8	fix(beam): GlobalCounters kernel count increasing when clearing l2 (#4598) ... * fix(beam): GlobalCounters kernel count increasing when clearing l2 * fix: removed the NOSTATS var by adding do_update_stats to Tensor.realize() * test(search): regression test for _time_program, should not increment kernel_count * fix(test_search): unused var and now properly checking when l2 is cleared * fix(test_search): added assert message * fix(test_search): now testing public beam api for kcount * ruff fixes --------- Co-authored-by: Léo Paillé <leo.paille@enseirb-matmeca.fr>	2024-05-19 10:03:47 -07:00
nimlgen	daf57af3eb	move tc to renderers (#4631) ... * move tc to renderers * missed import * fix typo * fix * fix imports * remove from tests * fix 4607 * nv emulate timestamp * time is int * correct time	2024-05-18 00:36:29 +03:00
chenyu	c86adabe15	time with real global buffers in search (#4621) ... * filter fake buffers in search * test that * update test	2024-05-17 12:36:23 -04:00
nimlgen	eb9689336e	nv mockgpu (#4600) ... * mockgpu nv * works * comment that out * fix merge * setup gpuocelot * install packages * not run all of them * passes * fix ci * almost * should pass * linter * linter 2 * try this? * ugn, not supported * ci * remove ticket from description * better descs	2024-05-15 23:46:08 +03:00
George Hotz	ff64bcab69	move graph/search to engine (#4596)	2024-05-14 23:12:59 -07:00