162 commits

Author	SHA1	Message	Date
chenyu	10c262ced8	update tests that use UOp.size (#15753)	2026-04-15 21:58:27 -04:00
qazal	96092d110c	fix process_replay Ops.BEAM [pr] (#15752)	2026-04-16 07:35:28 +09:00
Christopher Milan	be8005c5dc	DEV: secondary targets (#15748)	2026-04-15 17:26:20 -04:00
chenyu	507c02cecb	fix symbolic contiguous_view_offset (#15749) ... * fix symbolic contiguous_view_offset * flatten	2026-04-15 16:54:38 -04:00
Christopher Milan	1c36878008	DEV: suggest alternatives (#15732)	2026-04-14 23:42:32 -04:00
George Hotz	1ae6528bb6	move schedule into schedule (#15736) ... * move schedule into schedule * callify to root * sched docs	2026-04-15 11:03:25 +08:00
George Hotz	2450c8cba8	rename to callify + fix mypy (#15727) ... * rename to callify + fix mypy * update test	2026-04-14 23:43:19 +08:00
wozeparrot	2b8d303f75	allreduce in precast dtype (#15689)	2026-04-13 20:24:12 -07:00
George Hotz	5683126844	llm: support for tekken tokenizer (#15720)	2026-04-14 10:52:07 +08:00
chenyu	70883a6950	cat the stack to mixin (#15715)	2026-04-13 18:44:39 -04:00
qazal	905b8adc97	viz: cli and server cleanups (#15713) ... * update get_profile arg[0] * uop_to_json arg[0] * data is standalone in cli	2026-04-14 06:42:29 +09:00
Christopher Milan	d83707ec29	autogen: explicit types (#15679)	2026-04-13 16:54:39 -04:00
chenyu	ac41f15fc1	cumsum to mixin (#15712) ... built on top of getitem	2026-04-13 15:06:08 -04:00
chenyu	931d6cc62a	basic getitem to mixin (#15697) ... * basic getitem to mixin * cleanup * fix * cleanup	2026-04-13 13:04:36 -04:00
qazal	ac027055ef	viz: no global state (#15705) ... * start viz data * get_full_rewrites also moves * update ref_map * work * update consumers * cleaner cli * linter * cleanup tests * back * better * sqtt tests	2026-04-13 21:35:20 +09:00
George Hotz	4c1fb18a09	Revert "Revert "Tests for GatedDeltaNetBlock + fix multi after assign issue (…" (#15703) ... This reverts commit 0cec42db71.	2026-04-13 19:09:38 +08:00
George Hotz	0cec42db71	Revert "Tests for GatedDeltaNetBlock + fix multi after assign issue (#15700)" (#15702) ... This reverts commit 6f5d756282.	2026-04-13 19:06:44 +08:00
George Hotz	6f5d756282	Tests for GatedDeltaNetBlock + fix multi after assign issue (#15700) ... * broken after/assign test * test for GatedDeltaNet * better comments * fix issue 1 with multi kernel * fix 2 * fix * linter * public api + cleanup	2026-04-13 18:43:23 +08:00
chenyu	f7ff480fa6	start mixin getitem tests (#15695) ... goal is to make Tensor[idx].uop equal to Tensor.uop[idx]	2026-04-12 18:54:33 -04:00
George Hotz	b5a9465b13	llm: add support for moonlight (deepseek MLA) (#15466) ... * add gguf Q5_0 * it works * rebase * simpler test * class * less diff * dicts * normal names * simplify * this * simpler * work * work	2026-04-11 10:32:48 +08:00
b1tg	9ab1415937	llm: fix streaming UTF-8 decode (#15653)	2026-04-10 17:01:02 +08:00
Christopher Milan	dbc23e8a1b	move HCQ_VISIBLE_DEVICES into DEV (#15668)	2026-04-09 22:01:35 -04:00
Christopher Milan	d08c76d9cb	c.Struct cleanup (#15640)	2026-04-08 20:07:16 -04:00
qazal	dc6a51e44d	viz: add # of bytes to sdma (#15639) ... * viz: add # of bytes to sdma * update test_viz	2026-04-08 17:43:37 +09:00
Christopher Milan	19e96497ee	interface in DEV (#15620)	2026-04-06 19:59:28 -04:00
qazal	8ba58304f7	viz: reenable tests (#15626)	2026-04-07 07:52:44 +09:00
chenyu	2f7d085450	shared _normalize_indices for getitem (#15625) ... * shared _normalize_indices for getitem * list	2026-04-06 17:45:36 -04:00
chenyu	01b49c8647	support int operand for shifts (#15618) ... matches torch/jax, also symbolic rule to remove mask	2026-04-06 12:32:12 -04:00
Christopher Milan	645d45d968	DEV has arch (#15577) ... Co-authored-by: Comma Device <device@comma.ai>	2026-04-03 19:17:19 -04:00
Christopher Milan	0ed8d9271d	Renderers accept Target or nothing (#15590)	2026-04-03 01:09:41 -04:00
chenyu	1aa04eab08	simple CreationMixin (#15567) ... start with full_like, zeros_like, ones_like	2026-04-01 23:00:56 -04:00
Christopher Milan	6c67bd4c14	better error message when invalid renderer is specified (#15573)	2026-04-01 17:12:55 -04:00
b1tg	20497f2840	fold BIND to CONST when min==max (#15568)	2026-04-01 11:19:04 -04:00
chenyu	f5c0794df2	fix Tensor.const_like (#15565) ... used to always return a 0-d tensor, now returns an expanded Tensor based on self.shape and matches UOp	2026-04-01 08:35:19 -04:00
Christopher Milan	acf239e4d2	specify renderer in DEV, <dev>_<ren>=1 is deprecated (#15551)	2026-03-31 18:35:14 -04:00
chenyu	4ac2552642	improve ReduceMixin.all (#15544) ... use prod instead of min since `mul` lowered to `and` directly	2026-03-31 07:54:27 -04:00
chenyu	89ec22131a	tests to show double negation in min is not cancelled (#15543)	2026-03-31 06:59:13 -04:00
qazal	467c0af8aa	viz: skip flaky sever tests (#15538)	2026-03-31 17:20:30 +09:00
Christopher Milan	adbfd82d1d	DEV is ContextVar, setting Device.DEFAULT is deprecated (#15508)	2026-03-30 17:10:49 -04:00
chenyu	c0753ab62f	XOR simplifcation rules (#15512) ... x^-1 has good vmin/vmax, and x^y^y is x	2026-03-27 23:23:27 -04:00
nimlgen	0d6fc0f571	jit: graphing in uops (#15489) ... * jit: graphing as rewrite rule * f * +metal,cuda * x * cl * x * x * simpler * f * m * x * revert? * revert2 * back * back * t * x * m * x * c * x * l * x * comment * smaller * rv * x * x	2026-03-27 19:09:02 +03:00
chenyu	30ebbe7f17	few more fold valid tests (#15509) ... from remove CORRECT_DIVMOD_FOLDING attempt	2026-03-27 10:38:42 -04:00
chenyu	323fcefd7d	Revert "DEV is a ContextVar (#15505)" (#15506) ... This reverts commit fdb30cba96.	2026-03-27 02:22:40 -04:00
Christopher Milan	fdb30cba96	DEV is a ContextVar (#15505)	2026-03-27 00:57:09 -04:00
Christopher Milan	67a50fb738	move where on load with casts (#15492)	2026-03-26 22:11:27 -04: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
chenyu	7c8f992894	move EXPAND dtype cast back to gradient.py (#15481) ... only a concern for gradient, not mixin	2026-03-25 19:25:26 -04:00
chenyu	713b322e70	add weakint to promo_lattice (#15463) ... sits between bool and smallest int	2026-03-25 00:27:34 -04:00
George Hotz	fe2690399b	llm: support assistant prefill + refactor to TransformerConfig (#15457) ... * llm: support assistant prefill * refactor to ModelConfig * TransformerConfig * more	2026-03-25 10:50:48 +08:00
qazal	652bab8aad	viz: support nested track_rewrites (#15454) ... * simple test * stack active groups	2026-03-25 05:01:30 +09:00