730 commits

Author	SHA1	Message	Date
Sieds Lykles	4c3982c44e	Take sign out of mod (#11631) ... * Add rule and test * fix tests	2025-08-12 18:44:36 +02:00
George Hotz	ca41b5e38b	skip_0 in graph rewrite [pr] (#11627) ... * skip_0 in graph rewrite [pr] * no track_rewrites on test * use dict instead of set	2025-08-11 18:29:04 -07:00
George Hotz	996c907c0b	rewrite not ready + children machinery (#11607) ... * rewrite not ready + children machinery * it doesn't like track rewrites	2025-08-10 15:28:30 -07:00
qazal	960cc6533a	pass through name function args in track_rewrites (#11572)	2025-08-08 02:28:52 +03:00
George Hotz	82be8abfd2	move opt under codegen (#11569)	2025-08-07 14:19:17 -07:00
George Hotz	6ed2dfd187	delete the arange dim mismatch restriction (#11568) ... * delete the arange dim mismatch restriction * skip that test race	2025-08-07 13:46:17 -07:00
George Hotz	9764c6cdee	fix mismatch reduce, try 2 (#11560) ... * fix mismatch reduce, try 2 * fix heuristic * delete that test * don't start allowing ones	2025-08-07 07:57:58 -07:00
George Hotz	a1aa5670aa	Revert "fix mismatch reduce (#11547)" (#11549) ... This reverts commit 49d21a9055.	2025-08-06 22:43:15 -07:00
George Hotz	49d21a9055	fix mismatch reduce (#11547) ... * fix mismatch reduce * cleanups * fix shape * fix mypy * resolve	2025-08-06 21:12:51 -07:00
George Hotz	21570545d3	move view pushing to codegen, try 2 (#11534) ... * move view pushing to codegen, try 2 * fix up some linearizer tests * fix test search * fix test schedule * delete that test * fix test arange * fix a few tests * update tests * push views * ebs cleanup * fix local/reg * test and lint * fix more tests * test cleanups * skipped that one	2025-08-06 15:58:38 -07:00
George Hotz	80d9cced07	more test cleanups (#11544) ... * more test cleanups * revert that	2025-08-06 15:05:21 -07:00
qazal	846a2826ab	viz: remove TracingKey.fmt (#11482) ... * viz: remove TracingKey.fmt * remove from test too	2025-08-05 00:00:03 +03:00
leopf	4f0ee4e982	BPE tokenizer (#11415) ... * BPE works * refactor tok * oops * basic tests * fix eval * smaller diff * fix error * proper vocab decoding * use regex for splitting * escape ucatrange * full compat --------- Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>	2025-08-04 09:52:38 -07:00
chenyu	e0106b6b25	1/(x*c) -> (1/c)*(1/x) (#11491) ... example: 2*(2*a).reciprocal() -> a.reciprocal() # TODO: bounds for reciprocal # TODO: should z3 work?	2025-08-03 23:35:46 -04:00
chenyu	66be747908	few more dtype cast convinience methods (#11480)	2025-08-02 15:47:09 -04:00
chenyu	e22e5da9a5	move some test_dtype tests to unit (#11479)	2025-08-02 15:25:00 -04:00
qazal	fa66d9772d	viz: show const node when it's root (#11456)	2025-08-01 01:01:58 +03:00
chenyu	d5fc6af4a2	remove unused ShapeTracker.consecutive [pr] (#11426)	2025-07-29 18:36:19 -04:00
chenyu	88c338bfcc	add kernelize to keccak for each data block (#11370) ... * add kernelize to keccak for each data block test_long works now. this prevents internal uops from growing propotional to data length and eventually too deep * this? * hash stuff * gate test * mv	2025-07-25 16:07:20 -04:00
chenyu	82e6de7fc6	more keccak reference tests (#11329)	2025-07-23 22:06:39 -04:00
George Hotz	e14b4fefa5	ranges on store (#11334) ... * ranges on store * fix store spec * fix that * fix gates * fix tests * fix ptx	2025-07-22 21:00:50 -07:00
chenyu	4535908679	update keccak test_long (#11331) ... it should compare with arg "shake_128"	2025-07-22 16:08:01 -04:00
qazal	6668d6d241	fix word_wrap with newlines in input string [pr] (#11319)	2025-07-22 12:03:13 +03:00
George Hotz	842184a1ab	rename kernelize to schedule, try 2 (#11305)	2025-07-21 11:18:36 -07:00
wozeparrot	30ce16a424	feat: failing test for long keccak (#11292)	2025-07-21 12:49:23 -04:00
nimlgen	188ed38315	replace from_mv with lightweight mv_address (#11280)	2025-07-19 13:50:51 +03:00
quortus	924bc7c9ae	Fix test_uop_spec (#11259)	2025-07-16 11:02:31 +03:00
Alisher Zhubanyshev	4ef6b46b34	hcq: reduce launch overhead (#11193) ... * nv: improve mmio creation speed * add memoryview test * fix indents * move mv bench to `test_helpers`, remove comparison	2025-07-13 19:25:50 +03:00
chenyu	73caa5dd1b	remove Kernel.membufs [pr] (#11200)	2025-07-12 14:48:47 -04:00
qazal	d3ec63a5c3	viz: add base class for unittests (#11178)	2025-07-11 13:58:03 +03:00
chenyu	7db07e5f2c	don't narrow range of CAST on bool/unsigned (#11156)	2025-07-09 22:20:09 -04:00
George Hotz	53ae153404	tc should be in opt (#11148) ... * tc should be in opt [pr] * fix import	2025-07-09 14:12:21 -07:00
qazal	5c1d215b41	viz: add Graph stream (#11144) ... * viz: stack an event for the entire batch * multi * whitespace * work * multi graph, Graph gets its own row	2025-07-09 20:56:46 +03:00
qazal	3dfc0ff887	move cpu_profile and shared ProfileEvents from device.py to helpers [pr] (#11126) ... * move cpu_profile and shared ProfileEvents to helpers [pr] * TestProfiler.test_cpu_profile * update test_viz.py * TestProfiler.test_profile_multiops ordering, it's different streams now	2025-07-08 12:14:03 +03:00
Sieds Lykles	584fd6af5a	Fix division by zero and mask bug in add views (#11088) ... * merge view infinite loop test * adjust condition in `x//d -> x//(-d)*-1` * Fix division by zero in add views * adjust offset end * fix typo in comment * add target to test_merge_views_variable * fix view incorrectly being masked * ssimplify strides and offset of the new view to canonicalize * remove print in test --------- Co-authored-by: qazal <qazal.software@gmail.com>	2025-07-07 10:05:47 -07:00
qazal	81781dc12b	viz: renames and spacing changes to tracing (#11102)	2025-07-05 18:40:39 +03:00
qazal	4fcfaa0ef7	viz: switch to TracingKey (#11100) ... * viz: switch to TracingKey * tuple * order is name, keys, fmt * add test_tracing_key	2025-07-05 17:46:18 +03:00
qazal	3d8569f6d8	hotfix: infinite loop in tracking pattern matcher (#11094) ... * failing test * fix that * given matchers	2025-07-04 19:55:26 +03:00
qazal	b695e8c4d6	viz: remove support for naming with self (#11076)	2025-07-03 17:29:14 +03:00
Sieds Lykles	53985297bd	add test, fix rewrite rule and raise error on division by zero (#11073)	2025-07-03 08:25:06 -04:00
qazal	a919b8325b	add test_kernel_info (#11054) ... * add test_kernel_info * reorder	2025-07-02 19:48:12 +03:00
chenyu	4626e9c172	is_numpy_ndarray helper [pr] (#11050)	2025-07-02 09:12:53 -04:00
qazal	8b0871ac31	viz: test for no lockup on infinite loop (#11041) ... * viz: add test infinite loop fallback * assert * continue til the end * work * bring that back * fallback to nop	2025-07-01 17:44:20 +03:00
b1tg	fcbefde8f5	fix DiskDevice reuse (#11039) ... * fix DiskDevice reuse * fix mypy and DiskDevice.count * mypy * add test --------- Co-authored-by: b1tg <b1tg@users.noreply.github.com>	2025-07-01 10:29:21 -04:00
George Hotz	cccfe6b422	hotfix: test_no_inf_loop_bottom_up	2025-06-30 14:21:45 -07:00
George Hotz	b829331219	infinite loop detect in fixed_point_rewrite [pr] (#11038)	2025-06-30 08:57:29 -07:00
George Hotz	cb531dba42	detect infinite loop in graph rewrite [pr] (#11036)	2025-06-30 08:15:13 -07:00
qazal	2ea4737930	viz: fix newlines breaking label colors (#11030) ... * viz: fix newlines breaking label colors * TestViz.test_colored_label * TestWordWrap	2025-06-30 13:39:44 +03:00
George Hotz	5911b71404	early support for bidirectional pattern matcher (#11027) ... * early support for bidirectional pattern matcher * expose it and add a test * no bottom up arg there * disable flaky test	2025-06-29 16:54:07 -07:00
qazal	4c8d2a0383	buffer viz (#10960) ... * add mem_layout * ui * cleanup * work * debugLine work and expander * tooltip style * real expand device * wheel does one thing * diff * shows llama oom * add y axis * mypy chill * work * unittests for the memory layout	2025-06-28 21:50:32 +03:00