sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions Activity
11,218 Commits 336 Branches 15 Tags
gate_rewrite
Commit Graph
925 Commits
Author SHA1 Message Date
George HotzandGitHub c38b7684dc improve microbenchmarks (#13492) 
* improve microbenchmarks

* bugfix + ubench

* lil

* no src in const method
 2025-11-29 10:15:22 -08:00
qazalandGitHub 72ef533d9c tracing: use u32 for buffer args encoding (#13472) 2025-11-28 00:19:51 +08:00
George HotzandGitHub e4cd649ff0 remove kernelize to prepare for refactors (#13463) 
* remove kernelize to prepare for refactors

* less kernelize

* last test
 2025-11-26 14:18:50 -08:00
qazalandGitHub 7238df7a94 viz: cleanup sort_fn (#13454) 2025-11-26 04:10:10 +08:00
wozeparrotandGitHub 249553a119 tinyfs tweaks (#13444) 2025-11-24 18:07:32 -08:00
chenyuandGitHub cb29265f23 add test that shows the validhack regression with bad rewrite order (#13411) 2025-11-21 13:48:30 -05:00
chenyuandGitHub 0251a8e628 parse_valid minor cleanup [pr] (#13385) 
* stricter parse_valid [pr]

* not stricter

* no VCONST

* Revert "no VCONST"

This reverts commit 330dbdf4060562596febcbf970bda6051a35012f.
 2025-11-20 13:15:06 -05:00
George HotzandGitHub 986d113024 symbolic fuzz failure (#13367) 
* symbolic fuzz failure

* skip flaky test
 2025-11-19 14:21:08 -08:00
geohot 05ccc69248 Revert "merge to fold_divmod_general [p] (#13359)" 
This reverts commit 7711bbac7f.
 2025-11-19 14:18:09 -08:00
George HotzandGitHub 7711bbac7f merge to fold_divmod_general [p] (#13359) 
* merge to fold_divmod_general [p]

* merge more

* merge more

* merge more
 2025-11-19 11:37:45 -08:00
George HotzandGitHub 957cf717e7 Python speed (#13355) 
* skip process replay by default

* work on python speed

* fix names of rewrite rules

* fix that test
 2025-11-19 09:03:00 -08:00
sirhcmandGitHub a438c277de autogen tests for 3.14 (#13343) 2025-11-18 22:16:59 -05:00
George HotzandGitHub cabd4add48 more work parsing SQTT, separate VIZ/PROFILE (#13308) 
* more work parsing SQTT

* more minimal runner

* sep VIZ/PROFILE

* parse print new

* improve parser

* more filter

* that

* split them

* lil cleanup

* skip flaky test

* AQL in mmapeak
 2025-11-16 10:40:39 -08:00
nimlgenandGitHub c80d459d99 autogen: fix packed args structs (#13274) 
* autogen: fix packed args structs

* and test this
 2025-11-14 20:24:06 +08:00
geohot bcdfc109b5 hotfix: disable flaky test 2025-11-13 06:19:28 -08:00
George HotzandGitHub ab9fa964d8 DISABLE_COMPILER_CACHE -> CCACHE (#13234) 
* DISABLE_COMPILER_CACHE -> CCACHE

* Fix cachekey assignment in Compiler constructor
 2025-11-12 15:07:09 -08:00
qazalandGitHub 7a6853fa40 viz: show python callstack in the first graph (#13218) 2025-11-12 20:52:28 +08:00
sirhcmandGitHub 41a098a82d In-tree autogen: libc.py (#13217) 
* checkout changes from autogen branch

* parents

* pylint happy

* move sys to system in helpers.py

* typo

* typo
 2025-11-11 19:13:48 -08:00
qazalandGitHub bc55bc4849 cleanup test_viz profiler tests (#13221) 2025-11-12 03:46:48 +08:00
nimlgenandGitHub b8e48effcb device: no compilers message with reasons (#13146) 
* device: no compilers message with reasons

* typings

* mypy
 2025-11-07 23:01:45 +08:00
chenyuandGitHub bb8cf948f2 variation of (x%c)+(x//c)*c = x (#13135) 
when x is in the form of y//b, the idiv term might have combined
 2025-11-06 18:53:28 -05:00
George HotzandGitHub bcfe42937f move permute/flip/shrink to mixins (#13113) 
* move permute to mixins

* move more stuff

* two more

* fix local mypy

* fix tests

* fix shrink
 2025-11-05 14:14:15 -08:00
Sieds LyklesandGitHub 3dc593c536 add strip_params to pyrender (#13021) 
* add strip_params to pyrender

* update that one too

* strip_parens fix

* cleaner

* add test

* add some more tests

* cleaner strip_parens
 2025-10-31 14:15:56 +01:00
Sieds LyklesandGitHub 4c8362128b New symbolic renderer + strip parens (#13017) 
* new uop renderer

* better tester

* strip parens

* update tests

* split method check_uop_against_string

* use ctx.update instead of add_rendered method

* strip parens based on precedence

* update test

* new symbolic renderer

* add comment
 2025-10-30 16:41:32 +01:00
George HotzandGitHub 2da02f1ae1 add loads at the end (#12988) 
* add loads at the end

* simpler

* late load

* tests passing

* fix matvec

* spec test passes

* fix where on load

* fix abs2

* fix more tests
 2025-10-30 10:42:19 +08:00
Sieds LyklesandGitHub 70bce62c67 dont collapse possibly empty symbolic range (#12994) 
* dont collapse a symbolic range based on min/max

* refactor z3 renderer

* include sink explicitely instead of dtypes.void

* use dtype.scalar()
 2025-10-29 12:17:09 +01:00
Sieds LyklesandGitHub 9f39f6391c shared_codegen_spec and fix index spec (#12967) 
* split shared_codegen_spec and fix index

* add VCONST to program_spec and move index to shared_codegen_spec

* working ignore_oob=0

* cleanup

* fix spec

* undo that

* move barrier and special earlier

* fix more spec issues

* more updates

* remove special from program_spec

* cleanup and fixes

* move more to shared

* special is not in shared_spec

* some comments

* dont do bounds check there
 2025-10-29 09:14:11 +01:00
chenyuandGitHub e18922f111 limit AND const min max to ints [pr] (#12918) 2025-10-25 16:07:52 -04:00
George HotzandGitHub b4f6a2c7a3 add kernel spec (#12911) 
* add kernel spec

* fix kernel spec
 2025-10-25 11:49:20 +08:00
George HotzandGitHub 8a941d95a4 SPEC=2 is full spec, SPEC=1 is default (#12910) 
* SPEC=1 passes all tests

* just use SPEC, not __debug__
 2025-10-25 11:10:43 +08:00
wozeparrotandGitHub 9dac505565 variable bs keccak (#10731) 2025-10-23 14:10:21 -07:00
George HotzandGitHub 7762b3558b clean up the spec (#12868) 
* tighten up the spec

* move validate into a different file

* that moved to validate

* after(barr)
 2025-10-22 19:50:42 +08:00
George HotzandGitHub d711a4b933 delete old linearizer (#12834) 
* new linearizer with early endrange

* cleanups

* second stage removal

* not store

* do that later

* end cleanup

* fix globals

* end

* multi end

* fix ends earlier

* work

* do_merge_ends

* mini change

* range_gate

* fix cpu

* test fixups

* ranges on index

* not for ptx

* delete linearizer

* remove more junk

* delete that test

* we insert endif

* all ends
 2025-10-21 17:52:18 +08:00
George HotzandGitHub c780cd9abb new linearizer with early endrange (#12823) 
* new linearizer with early endrange

* cleanups

* second stage removal

* not store

* do that later

* end cleanup

* fix globals

* end

* multi end

* fix ends earlier

* work

* do_merge_ends

* mini change

* range_gate

* fix cpu

* test fixups

* ranges on index

* not for ptx
 2025-10-21 17:37:48 +08:00
qazalandGitHub 32af1ff84b viz graph drawing small cleanups (#12830) 
* viz graph drawing small cleanups

* str literal
 2025-10-21 15:51:32 +08:00
George HotzandGitHub 2e9082e0bc after op (#12801) 
* after op

* fix tests
 2025-10-20 12:27:56 +08:00
George HotzGitHubqazal
 		ba593f7b98 don't render index (#12796) 
* don't render index

* update to ignore_indexing

---------

Co-authored-by: qazal <[email protected]>
 2025-10-20 09:48:36 +08:00
qazalandGitHub c8ef4b60f6 viz: share match tracing and TINY device profiler (#12783) 
* set a default name for the traces

* set profile_matches + renames

* profile_matches test

* traces 4 steps total
 2025-10-19 14:30:07 +08:00
wozeparrotandGitHub 82f10cfe2e feat: assert on bufferview math (#12772) 2025-10-17 14:20:08 -07:00
qazalandGitHub 0160f034d6 viz: show display name for copy runners (#12761) 
* viz: show display name for copy runners

* more u32
 2025-10-17 16:59:51 +08:00
qazalandGitHub 253d32b065 viz: add metadata to buffer user list (#12758) 
* simple failing test

* encodings

* test passing

* key is deduped
 2025-10-17 16:28:54 +08:00
qazalandGitHub dfb8f9fc9e viz: annotate buffer mutability in the memory graph (#12750) 2025-10-17 11:53:02 +08:00
qazalandGitHub 533f18b22c viz: add trace data for inflight buffers (#12728) 
* viz: add trace data for inflight buffers

* add test_inflight_buf

* temp stores the keys

* update tests / use Tensor.ones
 2025-10-16 19:15:03 +08:00
George HotzandGitHub af4479c169 faster stable diffusion load (#12725) 
* faster stable diffusion load

* failing tests
 2025-10-16 18:31:59 +08:00
George HotzandGitHub 1d1e1d9d88 delete the ShapeTracker (#12720) 
* delete the ShapeTracker

* fix tests

* fix more

* fix gc test
 2025-10-16 15:36:22 +08:00
George HotzandGitHub 592e86f6f5 remove UOp.st (#12716) 
* remove UOp.st

* fix tests

* torch backend disable
 2025-10-16 14:44:09 +08:00
George HotzandGitHub 7c19db00f1 remove st from jit/split_reduceop (#12713) 
* remove st from jit

* fix by merging reshapes

* no st usage in rangeify

* hmm, stop early works

* fix speed regressions
 2025-10-16 12:50:58 +08:00
qazalandGitHub 069177c1be trace buffer producer and consumers (#12639) 
* trace buffer producer and consumers

* work

* generic colored util

* fix batched

* basic clicking works

* generic javascript that works for producer and consumers

* keep focused shape

* idle time

* timings for producer and consumers dedup

* from sd test

* tiny cleanups

* timeline

* work

* up to here

* assert

* list it

* work
 2025-10-16 11:11:31 +08:00
George HotzandGitHub 612e3d6143 replace mop arg with vectorized index (#12695) 
* replace mop arg with vectorized index

* tests passing

* better viz

* no compile4
 2025-10-15 20:50:06 +08:00
qazalandGitHub 768dc952de viz ui cleanups / renaming (#12691) 
* better viz names

* delete unused

* don't use opacity, it's multiplicative

* keep styles

* scrollbar coloring

* pyrender doesn't work here

beautiful_mnist r_64_16_32_36@lower all index dtypes
 2025-10-15 18:40:22 +08:00