sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions Activity
14,537 Commits 336 Branches 15 Tags
explicit-dev
Commit Graph
508 Commits
Author SHA1 Message Date
chenyuandGitHub c24da99d56 avg_pool2d, max_pool2d to mixin (#15903) 
* avg_pool2d, max_pool2d to mixin

* fix

* just dtype

* that
 2026-04-23 23:36:17 -04:00
chenyuandGitHub 08d9106c9f scatter_reduce and sparse_categorical_crossentropy to mixin (#15902) 
also use `.ne` to fix `# type: ignore[comparison-overlap]`
 2026-04-23 21:06:36 -04:00
chenyuandGitHub 8cc2c69e21 fix isclose mixin (#15898) 
use `.eq` instead of `==`
 2026-04-23 20:40:43 -04:00
chenyuandGitHub 782bc6aece broadcast in ElementwiseMixin.div [pr] (#15897) 2026-04-23 16:02:43 -04:00
chenyuandGitHub 11c197955b interpolate and cross_entropy to mixin (#15895) 2026-04-23 14:59:45 -04:00
chenyuandGitHub f0dbc68aa9 gather to mixin (#15891) 2026-04-23 14:00:57 -04:00
chenyuandGitHub 87223f870e logcumsumexp, argmax, argmin, sequential to mixin (#15890) 2026-04-23 12:10:42 -04:00
George HotzandGitHub 0c3260d5d9 rename VECTORIZE to STACK (#15880) 2026-04-23 10:43:42 +08:00
chenyuandGitHub 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
chenyuandGitHub 1fc4b3788c cummax/cummin to mixin (#15877) 2026-04-22 21:25:39 -04:00
chenyuandGitHub 684e95e1d4 UOp binary op broadcasts dtype (#15875) 
* UOp binary op broadcasts dtype

matches Tensor

* fix

* fix?
 2026-04-22 20:37:19 -04:00
sirhcmandGitHub b0dc95a390 AMX in arch, better docs (#15871) 2026-04-22 17:25:18 -04:00
nimlgenandGitHub 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
chenyuandGitHub b9e2bc619e simplify bool.cast() != const (#15874) 2026-04-22 17:08:09 -04:00
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
qazalandGitHub 719a7bdac5 viz: respect optional estimates in kernel info (#15867) 
* simple failing test

* unpack kernel info
 2026-04-22 14:24:48 +03:00
chenyuandGitHub 3821e442eb _one_hot_along_dim and one_hot to mixin (#15861) 2026-04-21 20:24:38 -04:00
chenyuandGitHub 75ee51a446 triu tril _tri to mixin (#15857) 2026-04-21 17:10:55 -04: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 d08b5d0a3b full to mixin (#15840) 
with unique_const
 2026-04-21 10:53:43 -04: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
chenyuandGitHub b01704444b einsum to ReduceMixin (#15833) 2026-04-20 11:49:24 -04: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
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
qazalandGitHub 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
chenyuandGitHub 0191cc73dc update arange range check (#15794) 
it was not checking negative steps correctly
 2026-04-17 16:07:50 -04:00
nimlgenandGitHub 23ca680a3a run_linear (#15784) 
* run_linear try 2

* x

* f

* tests

* ctx, cleaner

* r

* x
 2026-04-17 22:44:16 +03:00
qazalandGitHub 8fcaaede9a fix root cause of TestVizIntegration.test_link_sched_codegen flakiness (#15793) 2026-04-17 20:31:52 +03:00
qazalandGitHub afc3904e58 viz/cli: unit tests in CI (#15788) 
* simple failing test

* test stdout

* cleanup sqttmap
 2026-04-17 22:34:44 +09:00
wozeparrotandGitHub 9e60e4a7e7 llama: native fp8 (#15733) 2026-04-16 22:16:05 -07:00
George HotzandGitHub a9b6cfece0 refactor llm into files (#15780) 
* refactor llm into files

* chat.html

* tokenizer cleanup

* cleanup

* tests
 2026-04-17 12:33:11 +08:00
chenyuandGitHub 1fac03ce54 softmax and friends to mixin (#15778) 
with detach now
 2026-04-16 23:03:37 -04:00
George HotzandGitHub 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
qazalandGitHub 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
b1tgandGitHub 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
qazalandGitHub 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
MuzammilandGitHub 983a7bb576 exclude __del__ from TRACEMETA wrapping (#15747) 
Session-Id: 019d9234-2531-75a0-a252-f0302cd9931f
 2026-04-16 10:49:55 +08:00
chenyuandGitHub 8bd4fead26 UOp.size -> prod(max_shape) (#15755) 
and more test updates
 2026-04-15 22:41:30 -04:00
chenyuandGitHub 10c262ced8 update tests that use UOp.size (#15753) 2026-04-15 21:58:27 -04:00
qazalandGitHub 96092d110c fix process_replay Ops.BEAM [pr] (#15752) 2026-04-16 07:35:28 +09:00
sirhcmandGitHub be8005c5dc DEV: secondary targets (#15748) 2026-04-15 17:26:20 -04:00
chenyuandGitHub 507c02cecb fix symbolic contiguous_view_offset (#15749) 
* fix symbolic contiguous_view_offset

* flatten
 2026-04-15 16:54:38 -04:00