sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 8 Activity
14,300 Commits 336 Branches 15 Tags
simple_rangeify
Commit Graph
81 Commits
Author SHA1 Message Date
George HotzandGitHub 581bfdd94f merge track_rewrites and profile_matches into rewrite_group [PR] (#17420) 
* merge track_rewrites and profile_matches into rewrite_group

* bug

* flip ctx polarity
 2026-08-05 14:41:38 -07:00
chenyuandGitHub 59df317b12 use UOp.const to create new consts [PR] (#17368) 
replace arg won't work with ConstArg
 2026-08-02 13:31:55 -04:00
chenyuandGitHub b502fc1367 more const arg -> val (#17350) 2026-08-01 02:18:10 -04:00
chenyuandGitHub ad24750487 const(value, dtype) -> const(value).cast(dtype) in tests (#17335) 2026-07-31 13:24:31 -04:00
chenyuandGitHub 7f4dbb8090 remove shape= from UOp.const [PR] (#17331) 
inlined to const_like
 2026-07-31 11:18:59 -04:00
chenyuandGitHub 8e2f175542 const(dtype, b) -> const(b, dtype) [PR] (#17328) 
prep for dtype removal
 2026-07-31 09:46:37 -04:00
chenyuandGitHub ce500c1946 broadcast doesn't cast CONST [pr] (#17308) 
* broadcast doesn't cast CONST [pr]

it might change arg from python int to python float, but not cast the CONST, so we don't need or_casted in symbolic. we lower the weak dtype right before decomp

* fix
 2026-07-30 18:51:59 -04:00
chenyuandGitHub 0410c9325d make test/null follow the SPEC (#17010) 2026-07-13 14:01:41 -04:00
chenyuandGitHub 047a467bf9 delete UOp._stack and UOp.vectorize [PR] (#16988) 2026-07-11 15:26:55 -04:00
George HotzandGitHub d8fbbff260 EXPAND adds dims to the front (glm) (#16908) 
* EXPAND adds dims to the front (glm)

* more read images

* simpler + spec

* spec should use n instead of s'

* bump comma to 12
 2026-07-07 11:03:06 -07:00
qazalandGitHub f0a06fae4c viz: readable REWRITE_ERROR (#16903) 
* viz: readable REWRITE_ERROR

* traceback.format_exc handles this
 2026-07-07 17:25:42 +09:00
qazalandGitHub 9454e2ddbf viz: inf loop early cutoff (#16902) 
* pause on error

* reorder err

* simpler
 2026-07-07 17:00:45 +09:00
qazalandGitHub 795559b3d4 viz/cli: support passing only a start marker (#16802) 
* failing test

* support end interval
 2026-06-30 17:06:32 +09:00
chenyuandGitHub 3a9b87e48f remove Ops.DEVICE from Ops.PROGRAM [PR] (#16797) 2026-06-29 17:41:31 -04:00
nimlgenandGitHub 2ea8439d4d viz: fix enter_calls (#16793) 2026-06-29 17:10:26 +03:00
nimlgenandGitHub d9c65bd843 viz: fix walk rewrite (#16791) 2026-06-29 16:32:32 +03:00
qazalandGitHub daf8c8f509 viz: simpler test_link_sched_codegen, merge with the other codegen test (#16739) 
* simpler test_link_sched_codegen

* merge test_codegen_tracing in the other test

* include BEAM.value
 2026-06-25 15:02:00 +09:00
qazalandGitHub 695ffb3a08 viz: fix rendering custom Ops.PROGRAM source (#16738) 
* failing test

* fix

* simplify test
 2026-06-25 14:21:31 +09:00
George HotzandGitHub 4a4b6956df remove DEFINE_VAR from codebase (gpt) (#16666) 
* remove DEFINE_VAR from codebase

* junk

* remove junk
 2026-06-18 15:33:50 -07:00
George HotzandGitHub 5989d0b150 remove DEFINE_VAR try 2 (#16651) 
* remove DEFINE_VAR try 2

* param

* null index

* fix fuzzing

* fixes

* no gather neg params

* param is just Irreducible

* fixes

* skip stack

* need to filter slots there
 2026-06-18 12:34:25 -07:00
qazalandGitHub b753fb5e4c viz: view source working even if compile failed (#16657) 
* failing test

* hard

* ret_dict

* switch to _data for tests too

* update sqtt

* start work

* Ops.LINEAR looks good

* baseline with depth works

* support depth

* types

* @needs_tracked_pm

* update, marg can error too

* unwrap_or goes to many more places

* move things to soft_err

* soft_err everywhere needed

* diff cleanup

* use list

* rewrite it

* change

* update depth number

* small comment change
 2026-06-18 17:34:53 +09:00
qazalandGitHub 1cb6b88d37 viz: show contents of vconst (#16636) 
* failing test

* render vconst

* simpler test

* reorder
 2026-06-17 02:31:03 +09:00
qazalandGitHub 598cc13ad2 more readable null graph profile in VIZ (#16548) 
* more readable null graph profile in VIZ

* change

* fix flaky test
 2026-06-09 18:35:05 +09:00
chenyuandGitHub 4a8bf07a87 remove CONST(DEVICE) (#16506) 2026-06-04 11:29:46 -04:00
qazalandGitHub 81bc00c006 do not require clearing method_cache in viz tests (#16428) 
* update

* update test_dedup
 2026-05-29 18:12:34 +09:00
qazalandGitHub 54cfb794b8 viz: addrspace little colored box (#16427) 
* return addrspace

* layout

* render

* addrspace encodes color

* update colors

* in input_ast all are params are green

* update stroke
 2026-05-29 17:25:07 +09:00
George HotzandGitHub edca5df25a flip offset and shape in pad and shrink (#16414) 
* flip offset and shape in pad and shrink

* dumb test
 2026-05-28 11:58:19 -07:00
qazalandGitHub 846a809af7 viz: add +- toggle for hidden UOps (#16368) 
* first

* remove

* move src toggles to client side

* line

* update viz server tests

* remove those

* logic

* cleanup

* call matches

* fix const arg

* add labels

* keep changes

* the stack on movement ops hiding change

* structure

* rename to expandedNodes

* work

* test intention
 2026-05-26 22:31:54 +09:00
qazalandGitHub b73d2d17b9 viz/cli: add --interval (#16363) 
* interval support

* add test_interval

* llama uses interval
 2026-05-26 03:35:06 +09:00
qazalandGitHub 5cd0494b14 viz: canonicalize ast for schedule to codegen linking (#16154) 
* simple failing test

* always null device

* viz: canonicalize ast for schedule to codegen linking

* SCACHE
 2026-05-12 22:40:21 +09:00
qazalandGitHub 2dd84416bf viz/cli: schedule renderer (#16101) 
* simpler steps

* work

* work

* iterate

* faster

* better

* simplify more

* sys stdin

* less

* work

* work and mv

* better

* seen bufs

* all call graphs

* print query

* ux

* param to buffer / buffer_view

* work

* respect NO_COLOR in uop_to_json

* less

* render uops

* rm custom renderer

* call can't pyrender.

* unrelated diff

* assert

* 5
 2026-05-11 01:56:16 +09:00
qazalandGitHub 39ce780907 viz/cli: emit all runs of selected kernel, json fixes (#16124) 
* keep print

* --json in tests, sqtt --json err

* work

* import

* less

* line
 2026-05-10 21:45:51 +09:00
qazalandGitHub 51c7dafb0d split viz cli test helpers (#16123) 2026-05-10 19:42:24 +09:00
qazalandGitHub 795501e1da fix device in null graph events (#16053) 
* failing test

* fix compute

* fix sdma
 2026-05-06 07:44:08 +09:00
qazalandGitHub b1d88ebf02 viz/cli: aggregate flops in -t (#16031) 
* 38

* plumbing

* more flops

* flop/s and bytes/s

* arithmetic mean

* tests

* harmonic mean

* range

* better

* simplify

* fix prints

* no string parsing needed
 2026-05-04 17:35:02 +03:00
qazalandGitHub c02e390c2b viz: encode flops, mem and metadata in json (#16032) 
* gate print

* update everywhere to check path

* server encodes json

* ui changes

* cli changes

* tests never need regex

* no str replace

* update test_pipes

* remove that
 2026-05-04 23:06:18 +09:00
qazalandGitHub 9684334dfe viz: fix flops in graph, add null graph tracing (#16024) 
* min repro, todos

* null graph tracing

* work

* work

* work

* only test_flops

* exec points back

* first

* better

* integral timestamps maybe

* cleanup

* simpler, update NULL to use SDMA naming

* integration test

* sdma
 2026-05-03 22:32:44 +09:00
qazalandGitHub 7daf4b7d52 viz: split cli test (#16015) 
* viz: split cli test

* arg3 is msg
 2026-05-03 01:47:11 +09:00
qazalandGitHub 55915584e5 viz: fix cfg for emulated amd on the null device (#15976) 
* simple failing when i test it end to end

* pass

* linter

* assemble
 2026-04-30 05:18:09 +09:00
qazalandGitHub a37b605523 remove arch from asm kernel class (#15977) 
* rm arch from kernel

* update other tests

* update abstractions4.py
 2026-04-30 03:39:52 +09:00
nimlgenandGitHub 7787f76dcc get_runner -> get_runtime (#15967) 
* get_runner -> get_runtime

* do not use get_runner

* fix

* remove get_tunner

* remove

* fix

* x
 2026-04-29 18:29:49 +03:00
nimlgenandGitHub 4164666c72 programinfo (#15942) 
* programinfo

* fix

* m

* x

* x

* changes

* x

* fix

* rm
 2026-04-27 23:12:03 +03:00
qazalandGitHub 9a23de7d27 viz/cli: unify profile and rewrites, -s ALL default (#15931) 
* work

* workg

* better

* cleanup

* better defaults

* --ls

* better

* work

* update llama

* update
 2026-04-25 22:31:24 +09:00
nimlgenandGitHub 3c8a2db870 remove schedule() from tests batch 2 (#15923) 
* remove schedule() from tests batch 2

* batch 4
 2026-04-25 10:44:41 +03:00
sirhcmandGitHub 57fbaa3d49 amd: fallback to llvm when comgr is not available (#15914) 2026-04-24 23:30:16 -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
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
qazalandGitHub 719a7bdac5 viz: respect optional estimates in kernel info (#15867) 
* simple failing test

* unpack kernel info
 2026-04-22 14:24:48 +03: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
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