5,694 commits

Author	SHA1	Message	Date
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
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
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
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
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
chenyu	b01704444b	einsum to ReduceMixin (#15833)	2026-04-20 11:49:24 -04:00
nimlgen	c0d7135b5f	do not use jit_cache in test (#15823) ... * do not use jit_cache in test * fix	2026-04-20 11:45:17 +03:00
George Hotz	5819c0abed	fix gc in gguf (#15820) ... * fix gc in gguf * fix mypy	2026-04-20 10:15:03 +08:00
George Hotz	67ed4c4eb3	move gguf stuff from nn/state.py to llm/gguf.py (#15783) ... * move gguf stuff from nn/state.py to llm/gguf.py * docs	2026-04-20 09:41:43 +08:00
Kartik Vashishta	a1696e8413	objc: fix _classmethods_ dispatch flag (#14854) ... * objc: fix _classmethods_ dispatch flag * test: add objc _classmethods_ regression	2026-04-20 09:35:03 +08:00
oxrinz	f551a4bded	add threefry const folding (#15787) ... * prim threefry * test fix * clean test * cleanup * cleanup 2 * cleanup 3 * fix conflict markers in test_const_folding.py * update test * fix lint * use const instead of value for test	2026-04-20 09:30:03 +08:00
chenyu	2a5a6236ac	UOp.empty and UOp.empty_like (#15816) ... * UOp.empty and UOp.empty_like Tensor.empty and Tensor.empty_like use these, and removed _buffer_like * import line	2026-04-19 16:01:01 -04:00
qazal	c6d8753ee1	viz/cli: --json support, refine docs (#15528) ... * refine * remove * refine * keep * need to say this * back * feedback * feedback * json * dur_ms * et_ms * remove useless thing * docs * respect NO_COLOR * DEBUG also produces valid json	2026-04-19 21:53:38 +03:00
chenyu	5bdfd4883f	update test_assign (#15809) ... clean up old skips and update tests	2026-04-18 21:25:44 -04:00
Christopher Milan	6adf4c3cd9	MOCKGPU interfaces (#15796)	2026-04-17 21:56:29 -04:00
chenyu	8da308573f	update test_assign_changes_alt with clone (#15802)	2026-04-17 20:17:37 -04:00
qazal	2581985532	viz/cli: multi device profiler output, print markers (#15795) ... * yield * all devices * better * add unittests * markers like this * profile_markers work * less * update README * tiny and null	2026-04-17 23:40:10 +03:00
chenyu	0191cc73dc	update arange range check (#15794) ... it was not checking negative steps correctly	2026-04-17 16:07:50 -04:00
nimlgen	23ca680a3a	run_linear (#15784) ... * run_linear try 2 * x * f * tests * ctx, cleaner * r * x	2026-04-17 22:44:16 +03:00
qazal	8fcaaede9a	fix root cause of TestVizIntegration.test_link_sched_codegen flakiness (#15793)	2026-04-17 20:31:52 +03:00
qazal	afc3904e58	viz/cli: unit tests in CI (#15788) ... * simple failing test * test stdout * cleanup sqttmap	2026-04-17 22:34:44 +09:00
qazal	9f2a578e26	unskip TestCall.test_call_gemm_uop [pr] (#15786)	2026-04-17 16:18:51 +03:00
George Hotz	e1d13bc4fe	add GGUF IQ4_XS support (#15766) ... * add GGUF IQ4_XS support * gguf 21 * gguf 21 * use plus * ggml_common autogen for constant arrays * fix * ggml_common in autogen * inline	2026-04-17 14:43:39 +08:00
wozeparrot	9e60e4a7e7	llama: native fp8 (#15733)	2026-04-16 22:16:05 -07:00
George Hotz	a9b6cfece0	refactor llm into files (#15780) ... * refactor llm into files * chat.html * tokenizer cleanup * cleanup * tests	2026-04-17 12:33:11 +08:00
chenyu	1fac03ce54	softmax and friends to mixin (#15778) ... with detach now	2026-04-16 23:03:37 -04:00
George Hotz	ec00cefa5b	llm is the only app (#15779) ... * tinygrad/llm is the only app * upd pyproject * claude refs * scoping * min diff	2026-04-17 10:44:48 +08:00
Christopher Milan	9f4b7bed25	add pickled jit regression test (#15774)	2026-04-16 16:59:09 -04:00
qazal	12c653a743	remove opts arg in get_program, everything uses opts_to_apply [pr] (#15767) ... * check Ops.BEAM in process replay * remove opts from the get_program api * lint * simplify * cleanup	2026-04-16 22:42:43 +03:00