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
chenyu
f0c12a2004
another form of assign to itself ( #15770 )
2026-04-16 15:17:19 -04:00
b1tg
4e88d875ba
llm: glm 4.7 flash ( #15738 )
...
* glm 4.7
* test
* temperature, server enable_thinking
* --no-think
* remove think stuff
2026-04-16 22:42:04 +08:00
chenyu
d147e2a549
update test_nested_after_contiguous_store ( #15763 )
...
add kernel counts and some TODOs
2026-04-16 09:59:26 -04:00
qazal
126cda45f8
viz/cli: cleanups, add memory printer ( #15762 )
...
* simple repro
* use context
* work
* memory printer
* rm
* memory printer
* pylint
2026-04-16 22:44:47 +09:00
George Hotz
f57380cbc2
simplify GatedDeltaNetBlock using two state tensors ( #15704 )
...
* test double after
* simpler ssm
* no double test
2026-04-16 21:14:00 +08:00
George Hotz
d1cce7a476
put the ranges on store instead of after ( #15759 )
...
* put the ranges on store instead of after
* better assert
* fix stuff
* comment out slow rules i don't understand
* simpler rule
* closer
* return false for store
* fix loop
* only a few schedule failures remain
* remove stores to self
* all tests pass locally
* remove junk
* regression test and fix
* better test, bump broken torch count
* bugfix with regression test
* new fusion is better
2026-04-16 19:06:40 +08:00
George Hotz
d24466c844
CALL with return value is FUNCTION ( #15758 )
...
* CALL with return value is FUNCTION (GPT try)
* cleanups
2026-04-16 13:25:07 +08:00
chenyu
218d6b8988
delete old UOp.size [pr] ( #15756 )
2026-04-15 23:21:00 -04:00
Muzammil
983a7bb576
exclude __del__ from TRACEMETA wrapping ( #15747 )
...
Session-Id: 019d9234-2531-75a0-a252-f0302cd9931f
2026-04-16 10:49:55 +08:00
