13,471 commits

This branch

This branch

All branches

Author	SHA1	Message	Date
chenyu	782bc6aece	broadcast in ElementwiseMixin.div [pr] (#15897)	2026-04-23 16:02:43 -04:00
qazal	7745e05a2f	sqtt: update wave end packet names (#15896) ... * sqtt: update wave end packet names * update wavestart and emu	2026-04-24 04:21:22 +09:00
qazal	ee7644932b	viz/cli: -t default number (#15894) ... * viz/cli: accept one path argument * -t default * hm * only the -t change	2026-04-24 04:13:16 +09:00
chenyu	11c197955b	interpolate and cross_entropy to mixin (#15895)	2026-04-23 14:59:45 -04:00
chenyu	f0dbc68aa9	gather to mixin (#15891)	2026-04-23 14:00:57 -04:00
chenyu	87223f870e	logcumsumexp, argmax, argmin, sequential to mixin (#15890)	2026-04-23 12:10:42 -04:00
nimlgen	5cf4ad2fb6	fix resolve param (#15889)	2026-04-23 17:41:44 +03:00
nimlgen	e4696185bd	cleaner cuda graph (#15886)	2026-04-23 16:34:29 +03:00
wozeparrot	d3cbd781d9	llama: use fused norm mul quantize for w13 (#15878)	2026-04-22 21:27:41 -07:00
George Hotz	0c3260d5d9	rename VECTORIZE to STACK (#15880)	2026-04-23 10:43:42 +08:00
chenyu	7c9bc29e44	Tensor method raise if arg is on different device (#15879) ... instead of implicit `to`. this matches torch	2026-04-22 22:20:22 -04:00
chenyu	1fc4b3788c	cummax/cummin to mixin (#15877)	2026-04-22 21:25:39 -04:00
chenyu	684e95e1d4	UOp binary op broadcasts dtype (#15875) ... * UOp binary op broadcasts dtype matches Tensor * fix * fix?	2026-04-22 20:37:19 -04:00
Christopher Milan	b0dc95a390	AMX in arch, better docs (#15871)	2026-04-22 17:25:18 -04:00
nimlgen	e5891acab2	jit: precompile (#15848) ... * x * jit: precompile as sep step * x * s * x * x * x * ? * ? * x * x * viz * f * x * u * x * x	2026-04-23 00:23:32 +03:00
chenyu	b9e2bc619e	simplify bool.cast() != const (#15874)	2026-04-22 17:08:09 -04:00
nimlgen	2041945f4b	cuda graph to linear (#15870) ... * cuda graph to linear * fix * keep as old for now * x * x	2026-04-22 23:39:58 +03:00
chenyu	e9ebd03e86	update reduce_to_acc index dtype [pr] (#15873) ... index arg should have weakint dtype	2026-04-22 16:25:50 -04:00
chenyu	3c8daa9a75	update test_where_removal (#15872) ... don't use UOp.ufix for const_like, it will broadcast dtype soon	2026-04-22 14:56:37 -04:00
George Hotz	09ff3e1883	hotfix: add bytes back to llm	2026-04-23 00:46:27 +08:00
b1tg	af93a677ae	llm: glm 4.5 air (#15771) ... * llm: glm 4.5 air * clean * clean * remove gguf_size	2026-04-22 22:47:37 +08:00
qazal	719a7bdac5	viz: respect optional estimates in kernel info (#15867) ... * simple failing test * unpack kernel info	2026-04-22 14:24:48 +03:00
George Hotz	2d7fa58e61	fix shapes to match vecless (#15866) ... * fix shapes * need to simplify shapes	2026-04-22 18:27:46 +08:00
qazal	de8f58899e	move elf assembler to renderer (#15855) ... * move elf assembler to renderer * other	2026-04-22 19:00:36 +09:00
George Hotz	d4c344b7fd	hotfix: keep VCONST exclude in viz	2026-04-22 15:54:24 +08:00
wozeparrot	87378331e8	llama: fused mul quantize fp8 (#15863)	2026-04-21 20:58:37 -07:00
George Hotz	0560fa7b0f	add shape to range/special (#15862)	2026-04-22 11:15:02 +08:00
chenyu	3821e442eb	_one_hot_along_dim and one_hot to mixin (#15861)	2026-04-21 20:24:38 -04:00
chenyu	f911a63a6b	don't allow negative num_classes in one_hot (#15859) ... no auto infer num_classes, matches jax	2026-04-21 19:39:29 -04:00
Christopher Milan	697e7aa819	MOCK+AMD and MOCK+NV interfaces (#15858) ... MOCK+AMD is an alias for MOCKKFD+AMD, MOCKNVK+NV is renamed to MOCK+NV	2026-04-21 18:22:16 -04:00
chenyu	75ee51a446	triu tril _tri to mixin (#15857)	2026-04-21 17:10:55 -04:00
qazal	e36ff22538	fix dev syntax in emulated amd tests, skip test_tk (#15856) ... * fix dev syntax in emulated amd tests * skip test_tk	2026-04-21 23:47:29 +03:00
Christopher Milan	99a0debd62	Device.count() (#15842)	2026-04-21 16:46:38 -04:00
chenyu	1946ae8b51	linspace and eye to mixin (#15854)	2026-04-21 15:58:03 -04:00
qazal	0fbe0a6a99	viz/cli: ux tweaks (#15853) ... * viz/cli: rename to --json * st_ms, end confuses kimi * remove pickle spam * better * comment	2026-04-21 22:18:27 +03:00
chenyu	86ceb3bd6b	arange to mixin (#15852)	2026-04-21 13:00:19 -04:00
chenyu	420e4c4673	zeros, ones, invalids to mixin (#15850)	2026-04-21 11:53:08 -04:00
chenyu	9192c93b7e	Tensor.invalid -> Tesnor.invalids (#15849) ... matches ones and zeros, and to not share name with UOp.invalid	2026-04-21 11:19:51 -04:00
nimlgen	bfe28ee2ad	rm run_schedule (#15847)	2026-04-21 18:14:30 +03:00
chenyu	d08b5d0a3b	full to mixin (#15840) ... with unique_const	2026-04-21 10:53:43 -04:00
nimlgen	ae9b84d32f	rm beam uop (#15844)	2026-04-21 13:10:26 +03:00
nimlgen	01ac1c8c15	remove all run_schedule from tests (#15846)	2026-04-21 12:02:10 +03:00
qazal	f9655af2a3	viz/cli: move to tinygrad (#15835) ... * move cli * update imports * cleanup the readme * edit * work * details * python -m tinygrad.viz.cli * do not execv in non tty * option * lint * simpler * gemm pmc	2026-04-21 13:35:10 +09:00
Christopher Milan	1a8ba4cbd6	CPU renderers use arch (#15839)	2026-04-20 23:38:29 -04:00
chenyu	cabc347066	conv2d and conv_transpose2d to mixin (#15838) ... * conv2d and conv_transpose2d to mixin * cleanup	2026-04-20 18:10:06 -04:00
nimlgen	b8d3bf8970	run_linear in jit (#15827) ... * run_linear in jit * x * x * f * casts * ugh * f * x * x * simple	2026-04-20 23:03:30 +03:00
chenyu	e00cc8ae5e	split Tensor._conv2d_winograd (#15837)	2026-04-20 15:19:33 -04:00
chenyu	667b30b974	tensor pad arg cleanups (#15836)	2026-04-20 15:03:09 -04:00
chenyu	8eeb77a905	flat_to_grouped and resolve_pool_pads to helpers (#15834)	2026-04-20 14:03:35 -04:00
chenyu	b01704444b	einsum to ReduceMixin (#15833)	2026-04-20 11:49:24 -04:00