Commit graph

13,471 commits

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