nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity
13,053 Commits 334 Branches 15 Tags
3c8daa9a75fccdbfdb4ef3d523e61dc616c8bfab
Commit Graph
13053 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
chenyuandGitHub 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
geohot 09ff3e1883 hotfix: add bytes back to llm 2026-04-23 00:46:27 +08:00
b1tgandGitHub 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
qazalandGitHub 719a7bdac5 viz: respect optional estimates in kernel info (#15867) 
* simple failing test

* unpack kernel info
 2026-04-22 14:24:48 +03:00
George HotzandGitHub 2d7fa58e61 fix shapes to match vecless (#15866) 
* fix shapes

* need to simplify shapes
 2026-04-22 18:27:46 +08:00
qazalandGitHub de8f58899e move elf assembler to renderer (#15855) 
* move elf assembler to renderer

* other
 2026-04-22 19:00:36 +09:00
geohot d4c344b7fd hotfix: keep VCONST exclude in viz 2026-04-22 15:54:24 +08:00
wozeparrotandGitHub 87378331e8 llama: fused mul quantize fp8 (#15863) 2026-04-21 20:58:37 -07:00
George HotzandGitHub 0560fa7b0f add shape to range/special (#15862) 2026-04-22 11:15:02 +08:00
chenyuandGitHub 3821e442eb _one_hot_along_dim and one_hot to mixin (#15861) 2026-04-21 20:24:38 -04:00
chenyuandGitHub 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
sirhcmandGitHub 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
chenyuandGitHub 75ee51a446 triu tril _tri to mixin (#15857) 2026-04-21 17:10:55 -04:00
qazalandGitHub 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
sirhcmandGitHub 99a0debd62 Device.count() (#15842) 2026-04-21 16:46:38 -04:00
chenyuandGitHub 1946ae8b51 linspace and eye to mixin (#15854) 2026-04-21 15:58:03 -04:00
qazalandGitHub 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
chenyuandGitHub 86ceb3bd6b arange to mixin (#15852) 2026-04-21 13:00:19 -04:00
chenyuandGitHub 420e4c4673 zeros, ones, invalids to mixin (#15850) 2026-04-21 11:53:08 -04:00
chenyuandGitHub 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
nimlgenandGitHub bfe28ee2ad rm run_schedule (#15847) 2026-04-21 18:14:30 +03:00
chenyuandGitHub d08b5d0a3b full to mixin (#15840) 
with unique_const
 2026-04-21 10:53:43 -04:00
nimlgenandGitHub ae9b84d32f rm beam uop (#15844) 2026-04-21 13:10:26 +03:00
nimlgenandGitHub 01ac1c8c15 remove all run_schedule from tests (#15846) 2026-04-21 12:02:10 +03:00
qazalandGitHub 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
sirhcmandGitHub 1a8ba4cbd6 CPU renderers use arch (#15839) 2026-04-20 23:38:29 -04:00
chenyuandGitHub cabc347066 conv2d and conv_transpose2d to mixin (#15838) 
* conv2d and conv_transpose2d to mixin

* cleanup
 2026-04-20 18:10:06 -04:00
nimlgenandGitHub 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
chenyuandGitHub e00cc8ae5e split Tensor._conv2d_winograd (#15837) 2026-04-20 15:19:33 -04:00
chenyuandGitHub 667b30b974 tensor pad arg cleanups (#15836) 2026-04-20 15:03:09 -04:00
chenyuandGitHub 8eeb77a905 flat_to_grouped and resolve_pool_pads to helpers (#15834) 2026-04-20 14:03:35 -04:00
chenyuandGitHub b01704444b einsum to ReduceMixin (#15833) 2026-04-20 11:49:24 -04:00
chenyuandGitHub 3a557016cb delete UOp.get_consumer_map [pr] (#15832) 
not used
 2026-04-20 10:57:42 -04:00
chenyuandGitHub 04e8dbd7f8 remove getitem check in get_shape (#15830) 
not needed
 2026-04-20 10:40:46 -04:00
chenyuandGitHub 72ecc61ca8 use more UOp method [pr] (#15821) 
instead of constructing UOp directly
 2026-04-20 09:17:56 -04:00
qazalandGitHub 601b9d3f59 viz/cli: dedup DEBUG=3 pyrender (#15826) 2026-04-20 19:29:09 +09:00
ayanhanandGitHub 80c7327e0f resolve Metal ARC FIXME with explanation comment (#13688) 2026-04-20 17:10:37 +08:00
nimlgenandGitHub 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 HotzandGitHub 5819c0abed fix gc in gguf (#15820) 
* fix gc in gguf

* fix mypy
 2026-04-20 10:15:03 +08:00
George HotzandGitHub 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
chenyuandGitHub 538841d1f2 remove_tags and _remove_all_tags are the same [pr] (#15819) 
also other small UOp method cleanups
 2026-04-19 21:37:49 -04:00
Kartik VashishtaandGitHub 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
oxrinzandGitHub 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
qazalandGitHub b05b1010bf viz/cli: ux cleanups, show user python (#15817) 
* small fixes

* print python trace

* jsonl

* cleanup fmt, fix tqdm

* print mode

* types

* less

* keep those

* fix

* everyone can print json

* pmc p2
 2026-04-20 03:50:48 +03:00
chenyuandGitHub 8b87b3522a more UOp empty cleanups [pr] (#15818) 2026-04-19 19:48:36 -04:00
chenyuandGitHub 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
qazalandGitHub 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
chenyuandGitHub 50a7b82372 merge untag_and_append and append_after [pr] (#15815) 
reads cleaner
 2026-04-19 13:13:26 -04:00
chenyuandGitHub cace07c87a clean up untag_and_append [pr] (#15812) 
replace_uop does not change, and ret.op is always AFTER
 2026-04-19 11:23:59 -04:00
wozeparrotandGitHub f28ea84de2 llama: fused silu fp8 amax (#15798) 
* llama: combined w13

* llama: fused swiglu+fp8

* llama: fix amax interleaving

* llama: don't need seperate matmul
 2026-04-19 12:03:55 +08:00