710 commits

Author	SHA1	Message	Date
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
qazal	a39343e39f	viz: move timeline layout to python (#10998) ... * viz: move timeline layout to python * DevEvent has a device and a name	2025-06-27 13:06:00 +03:00
qazal	1127302c46	move perfetto to extra (#10994) ... * move perfetto to extra * update TestViz and fix tests * remove perfetto.html from viz directory * work * mypy	2025-06-27 01:53:54 +03:00
Ignacio Sica	21f1c4cc09	remove some linearize calls from tests [pr] (#10978) ... * remove some linearize calls from tests speed_compare_cuda_ptx test_uop_spec test_linearizer test_uops test_winograd * more clear assert message	2025-06-25 12:37:17 -07: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
Sieds Lykles	61dad3740f	fix min_max and add test (#10952)	2025-06-24 09:33:26 -04:00
qazal	f41c28a048	update test_tensor_uop_representation comments [pr] (#10946) ... These comments can update to match new tinygrad.	2025-06-24 10:47:09 +03:00
Harsh Natuskar	79d7cdd9ba	Fix device (#10929) ... * fix: pkg * better * added test * less lines	2025-06-23 15:30:19 -07:00
qazal	ac39f27ae6	viz: non blocking UOp tracing (#10913) ... * viz: non blocking UOp tracing * u.arg * no if Ops.KENREL * drop replace * switch to weakref.WeakKeyDictionary * back * remove ram usage skips, viz works here * cache on reconstruct	2025-06-23 19:59:28 +03:00
qazal	9201224e0b	viz: remove Kernel check [pr] (#10920) ... * viz: remove Kernel check [pr] * TestVizIntegration * test/unit allows opening of devices * kernel -> Kernel	2025-06-22 20:47:54 +03:00
qazal	c7ec913210	viz: cleanup unit tests (#10909) ... * cleanup test_viz * tree view	2025-06-21 12:35:09 +03:00
George Hotz	1ce63f8d04	move functions to view and update docs [pr] (#10904) ... * move functions to view and update docs [pr] * move quantize	2025-06-20 16:47:58 -07: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	fc9f883870	if upat returns self, it's none (#10898) ... * if upat returns self, it's none * fix pm tests	2025-06-20 12:11:19 -07:00
simone-pietro	e94ac6e20c	Cast ptr to int in test_from_mv_to_mv (#10876) ... * Cast ptr to int in test_from_mv_to_mv * Add type hints for from_mv	2025-06-20 14:52:34 +03:00
simone-pietro	36f01411a2	Pass list to block_reorder in test_loads (#10881)	2025-06-19 09:49:45 -04:00
qazal	00d0071b36	simpler viz naming [pr] (#10874) ... * simpler viz naming [pr] * n2	2025-06-19 12:10:47 +03:00
qazal	5839542fc8	viz: one name arg in track_rewrites [pr] (#10873) ... * viz: one name arg in track_rewrites [pr] * other test	2025-06-19 03:34:56 +03:00
simone-pietro	d8cea1a279	Change rate to int in test_tqdm (#10848)	2025-06-18 08:40:43 -04:00
qazal	8b879b0314	merge TestTensorUOpSpec with the other spec unittests [pr] (#10860) ... * merge TestTensorUOpSpec with the other spec unittests [pr] * rename to test_uop_spec	2025-06-18 12:12:08 +03:00