123 commits

Author	SHA1	Message	Date
nimlgen	ae9b84d32f	rm beam uop (#15844)	2026-04-21 13:10:26 +03:00
nimlgen	057dc173ab	beam uop (#15660) ... * beam as uop * x	2026-04-09 19:13:03 +03:00
Christopher Milan	bc180a963c	deprecate <dev>=1 in favor of DEV=<dev> (#15467) ... * start work on target * add test * update actions to use DEV * update docs * update readmes * tests need that too * update example * update tests (comments) * fix that test * ruff * mypy * oops * remove getenvs * don't add Target yet * and the test * lint * and docs * more stuff * assert * few more fixes * test assert	2026-03-26 03:48:03 -04: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	d59e6e7a37	move more tests to test/null, split some existing ones (#14512) ... * move more tests to test/null, split some existing ones * null work * null work * move more * fixes * move PIL * PIL in CLIP * don't move that	2026-02-03 20:20:20 +08:00
chenyu	66d2b02f11	delete files that depends on extra.optimization.helpers (#14499)	2026-02-02 13:33:33 -05:00
chenyu	7e41da1ae8	fix generate_dataset.sh (#14324) ... added `set -e` so wrong pathes would fail the script, then fixed the path	2026-01-24 16:47:10 -05:00
b1tg	24723327ac	fix tc_up in search (#13438) ... * tensor_core is missing from Scheduler * test upcast max --------- Co-authored-by: chenyu <chenyu@fastmail.com>	2026-01-01 10:25:08 -05:00
chenyu	7f1d41c9f9	delete files that import ShapeTracker (#13805)	2025-12-22 15:54:18 -05:00
chenyu	c8dfd10257	ShapeTracker.real_strides -> is_expanded [pr] (#12579) ... only keep the used part	2025-10-09 22:52:45 -04:00
Sieds Lykles	cc038b31b6	Shrink instead of reshape to unregister symbolic (#12241) ... * Slice to unbind symbolic * use vmax for now * assert shape in reshape is valid * update test_symbolic_ops to use shrink instead of reshape * remove infer_with_bound_values for npw * symbolic output doesnt have symbolic strides * symbolic jit tests use shrink to unregister symbolic * update test * update more tests * wrap vmax in int() * only create a new st if the store is not an assigne * unwrap st * comments	2025-09-19 06:04:35 +02:00
chenyu	0e266f376c	ops_gpu -> ops_cl (#12103)	2025-09-10 15:15:48 -04:00
Sieds Lykles	c6c16b2946	var_vals uses str for var (#12011) ... * var_vals is str,int * remove imports * remove print * fix test * change var_vals in hcq * update test_hcq * fix multitensor _device_num var * fix syminfer test * shorten line * p.vars stays list[Variable] * shorten line * vars is back to tuple[Variable, ...] * change var_vals in extra * change var_vals from shapetracker * var_vals is str:int * fix signature	2025-09-06 04:16:12 +02: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
Ben Waldron	ea1be2e4cd	[bounty] Remove using reshape to register symbolic shape (#11771) ... * Modify tests and start work towards removing symbolic reshape * Refactor symbolic reshape * fix small error * much cleaner + fix more tests * Can remove this now * Update test_symbolic_ops and test_tiny * Couple more tests * Unused import * More tests and add EXPAND to Tensor.empty * Fix test beam search * all int * Fix rangeify by adding shrink * Remove OOB check and so fix test_symbolic_jit * test_symbolic_jit doesn't need OOB Context anymore either * Should remove that test now * Cleanups part 1 * fix linters * Final cleanups * Don't reassign inside for loop --------- Co-authored-by: chenyu <chenyu@fastmail.com>	2025-08-28 12:30:49 -04:00
George Hotz	82be8abfd2	move opt under codegen (#11569)	2025-08-07 14:19:17 -07:00
chenyu	a0438012af	remove Kernel.get_program [pr] (#11203)	2025-07-12 20:50:29 -04:00
chenyu	425d5f55c4	generate kernel dataset and upload artifact (#11063)	2025-07-02 17:21:25 -04:00
chenyu	a6485d00c8	very tiny generate_dataset (#11013) ... one minute to gen on my mac	2025-06-27 17:10:45 -04:00
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
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
George Hotz	92678e59ee	move kernel to opt (#10899)	2025-06-20 15:22:28 -07:00
chenyu	5e7ad70aae	don't run linearize().uop tests in get_action_space test (#10766) ... * don't run linearize().uop tests in get_action_space test this part takes 2 minutes in CI and has nothing to do with action space. also not sure if the "for some reason" comment is still relevant * -n=auto test/models	2025-06-10 17:23:53 -04:00
George Hotz	b3b43a82c4	remove Tensor.no_grad, it's meaningless now [pr] (#10556)	2025-05-28 22:20:02 -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
Sieds Lykles	e75be6eafc	[bounty] [pr] index validation with z3 (#9981) ... * index validation with z3 * Change comment * toposort -> toposort() --------- Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>	2025-04-24 08:06:08 -04: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	f5256e0020	Kernel.apply_opts [pr] (#9917) ... * Kernel.apply_opts [pr] updated all `for opt in`. also updated a few test_liinearizer tests to not implcitly depend on hand_coded_optimization * not you yet	2025-04-17 08:00:56 -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	1fda98d14f	fix import time_linearizer [pr] (#9118) ... only test that used it was skipped in CI due to being slow	2025-02-15 21:33:28 -05: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
qazal	60503c8621	use CAPTURE_PROCESS_REPLAY=1 in CI [pr] (#8564)	2025-01-11 06:03:48 -05:00
qazal	07b6d5cf63	assign early folding (#8093) ... * assign early folding [pr] * move to to_si * - * fix generate_dataset * diff too big * no recreation, no diff * gzip * new sops from tiny10 * final try	2024-12-07 17:02:55 +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
qazal	e84d089ef1	delete ReduceOps, only use REDUCE_AXIS (#7667)	2024-11-13 19:04:27 +08:00
qazal	e07d2d0966	skip TestBeamSearch.test_large_ast (#7652)	2024-11-12 20:52:22 +08:00
chenyu	e7b18cf5c0	fix load_worlds filter_novariable (#7564) ... filter based on "DEFINE_VAR" instead of "Variable". also added a unit test to make sure dataset includes image and variable kernels	2024-11-05 16:06:39 -05:00
chenyu	207bca6cea	set PAGE_SIZE=1 and generate new dataset (#7559) ... 13080 rows in total. both generating and loading this are pretty broken now. filters are wrong for example	2024-11-05 11:25:01 -05:00
chenyu	7581a57aac	show the actual dataset size in error message (#7557)	2024-11-05 09:16:30 -05:00
chenyu	0db5f52b2a	check datasets/sops.gz size to be > 5000 (#7555) ... it has > 12000 rows now, but it depends on the backend that generates these so setting a lower but meaningful threshold	2024-11-05 09:03:19 -05:00
George Hotz	c8bf09b7d4	s/UOps/Ops (#7500) ... * s/UOps/Ops [pr] * fix	2024-11-03 11:26:10 +08:00
qazal	8ff6514ba3	delete extra/ops.py [pr] (#7072)	2024-10-15 22:14:21 +03:00
chenyu	bd8ecf7fd6	remove NumNode (#7035)	2024-10-13 16:42:19 -04:00
chenyu	c4c806a210	generate new kernel dataset (#7034) ... * generate new kernel dataset pre req to remove NumNode ``` extra/optimization/generate_dataset.sh gzip -k /tmp/sops mv /tmp/sops.gz extra/datasets/ ``` * fix var range in fuzz_linearizer	2024-10-13 16:19:41 -04:00
qazal	13846930cd	hotfix: extract_dataset.py (#7029)	2024-10-13 11:18:23 +03:00
George Hotz	a71bb09ec3	remove symbolic file [pr] (#7012)	2024-10-12 18:44:44 +08:00
George Hotz	5ae2de9845	UOp.variable (#7010) ... * UOp.variable [pr] * fix tests * clean * improve name rendering * last bug	2024-10-12 18:20:44 +08:00
George Hotz	a0cb16ac61	node cleanup + local metal test speed [pr] (#6880) ... * node cleanup [pr] * fix tests, including the double one on metal * no time tqdm tests	2024-10-04 18:14:23 +08:00
qazal	9295bc0189	viz more work [run_process_replay] (#6568) ... * infra * found it * real work * bring those back * cleanup test_viz * comment that out	2024-09-17 19:27:09 +08:00