nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity
14,154 Commits 334 Branches 15 Tags
6a8bb39f3d5c19f7fa971c779fc2b281014cffc6
Commit Graph
66 Commits
Author SHA1 Message Date
qazalandGitHub e684fcc689 pm_reduce_collapse fix for re enabling stack for cat of same shape (codex) (#17251) 
* change

* stack cat

* remove nop

* change
 2026-07-29 14:47:28 +09:00
George HotzandGitHub 7d48926293 switch _device_num to AxisType.DEVICE range (kimi) (#17252) 
* switch _device_num to AxisType.DEVICE range (kimi)

* improve

* don't exclude device for ranges

* current multi ends the range
 2026-07-28 13:52:47 -07:00
chenyuandGitHub c9e11544df delete explicit casts [pr] (#17255) 
handled by broadcast
 2026-07-28 13:57:13 -04:00
George HotzandGitHub 39924387b1 LOOP is srcless RANGE (kimi) (#17129) 
* LOOP is srcless RANGE (kimi)

* upd

* cleanups

* cleanups

* we don't float anymore
 2026-07-21 18:41:31 -07:00
nimlgenGitHubgeohot
 		b764599d87 add Ops.LOOP + conditional Ops.END (kimi) (#17117) 
* add Ops.LOOP + conditional Ops.END (kimi)

* c

* x

---------

Co-authored-by: George Hotz <[email protected]>
 2026-07-21 22:54:14 +03:00
chenyuandGitHub 13ca9bd8a6 remove dtypes.index again (#17104) 
also reverted some dtype change, the split made things needlessly complicated
 2026-07-20 20:30:04 -04:00
chenyuandGitHub ba2c68b1ed start dtypes.index [PR] (#16949) 2026-07-09 15:12:23 -04:00
nimlgenandGitHub c9f6b2e42b hcq2: prereq for linker (#16945) 2026-07-09 17:21:21 +03:00
George HotzandGitHub e4d0d634d4 remove a bunch of dtype.count [PR] (#16929) 
* remove a bunch of dtype.count

* remove nonsense rules

* rm dead code
 2026-07-08 09:34:46 -07:00
George HotzandGitHub 0035bb6fa8 only front reduce (#16901) 
* only front reduce

* lint + spec

* that should fail

* meh
 2026-07-07 09:55:07 -07:00
George HotzandGitHub 41d6731bfd reduce removes ones (#16847) 
* reduce removes ones

* test changes

* lil clean
 2026-07-03 13:10:08 -07:00
chenyuandGitHub 59de2fa3d3 combine pm_reduce_collapse rules [PR] (#16787) 2026-06-28 13:10:13 -04:00
George HotzandGitHub 649971f02a remove DEFINE_LOCAL and DEFINE_REG (gpt) (#16673) 
* remove define_local and define_reg (gpt)

* fix precommit

* cleanups

* regalloc fix

* cleanups 2
 2026-06-19 10:07:50 -07: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
George HotzandGitHub 55515747b7 Remove Ops.VCONST (#16267) 
* start removing vconst

* remove a lot of vconst

* const folding + strict ordering

* update tests

* spec from minigen

* move that
 2026-05-19 16:35:24 -07:00
chenyuandGitHub 34fe37d64e use FLOORDIV and FLOORMOD (#16048) 
* use FLOORDIV and FLOORMOD

also removed CORRECT_DIVMOD_FOLDING

* fix

* Revert "fix"

This reverts commit 86af33b88ef31943c61e67189b072eca4896409a.

* fix

* fix
 2026-05-05 18:32:54 -04:00
George HotzandGitHub 4a2e1f1076 STORE doesn't have ranges anymore (#16019) 
* STORE doesn't have ranges anymore

* fix
 2026-05-01 15:00:27 -07:00
George HotzandGitHub 5f441ecffc unify reduce + reduce_axis (#15973) 
* unify reduce + reduce_axis

* fix all tests

* lil cleanups
 2026-04-29 10:29:56 -07:00
chenyuandGitHub 72ecc61ca8 use more UOp method [pr] (#15821) 
instead of constructing UOp directly
 2026-04-20 09:17:56 -04:00
sirhcmandGitHub 9e0cc5c6ae create image buffers in late codegen (#15493) 2026-03-27 04:50:53 -04:00
George HotzandGitHub 1a2a203f48 add wmma support to amd_copy_matmul (#15384) 
* add wmma support to amd_copy_matmul

* 15 TFLOPS and merged

* unify

* simpler

* simpler

* simpler

* cleanups

* TM/TN is the full regs

* comments

* WAVES_PER_SH + SQTT_EVENT

* Add WAVERDY support

* no split warp

* 3 range
 2026-03-20 19:02:19 +08:00
chenyuandGitHub da1700e16b dtypes.index -> dtypes.weakint (#15377) 2026-03-20 01:08:46 -04:00
sirhcmandGitHub d753c5d7e5 IMAGE=1 image_conv2d pads for bank conflicts (#15252) 2026-03-14 07:59:16 -04:00
sirhcmandGitHub dabdc986df shrink guarded ranges, try 2 (#15272) 2026-03-14 04:24:05 -04:00
sirhcmandGitHub 7cf4b16c91 Revert "shrink guarded ranges" (#15271) 2026-03-14 03:44:38 -04:00
sirhcmandGitHub d9951e2f8e shrink guarded ranges (#15263) 2026-03-14 03:38:48 -04:00
chenyuandGitHub 605b37c03f use backward_slice in count_divmod [pr] (#15178) 2026-03-06 14:03:53 -05:00
George HotzandGitHub dd2de4f838 rename all DEFINE_GLOBAL to PARAM (#14511) 2026-02-03 15:09:38 +08:00
chenyuandGitHub 3c5845e8a5 remove cut_store_range (#14505) 
special scheduling for CPU
 2026-02-02 21:58:36 -05:00
chenyuandGitHub 31706bf6bc add few more types [pr] (#14425) 2026-01-29 14:04:09 -05:00
chenyuandGitHub 51398edf9c fix indirect import (#13958) 
also deleted old external tests
 2026-01-01 14:22:45 -05:00
chenyuandGitHub 21aac568fd limit lift x*y out of reduce to int [pr] (#13535) 2025-12-02 16:11:45 -05:00
Sieds LyklesandGitHub 114bb94c55 Fix load collapse MAX to ADD (#13406) 
* add Ops.ADD to pattern

* add test
 2025-11-21 12:26:14 +01:00
chenyuandGitHub 41e45c20ff minor stuff reading the printed code [pr] (#13177) 2025-11-09 00:58:51 -05:00
chenyuandGitHub ca17718b6d remove symbolic_flat (#13083) 
* remove symbolic_flat

some kernels are different but sometimes it's better so not clear, will merge as long as benchmark passes

* test_location
 2025-11-03 17:25:21 -05:00
Sieds LyklesandGitHub 885b6dea9e multiple reduce range arange folding (#13047) 
* multi reduce arange folding

* add test

* cvar to var

* add circular_pad_bw test
 2025-11-01 22:11:26 +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 5ce8a1d2f2 Merge adjacent try all permutations for reduce (#12972) 2025-10-29 05:04:54 +01:00
Sieds LyklesandGitHub e110f4632a split cat (on cpu) (#12864) 
* split ranges but only on cpu

* except KernelOptError for threads

* use GROUP and END

* no more flatten_range needed

* remove noop end

* always process replay for openpilot

* update test

* skip test

* fix in outs calculation

With the new linearizer the toposort is a problem, this matches the spec
now

* undo that
 2025-10-28 07:55:19 +01:00
George HotzandGitHub 6b35467f53 stores don't end ranges (#12902) 
* early endrange

* bugfixes
 2025-10-24 23:05:03 +08:00
Sieds LyklesandGitHub c1db62ff7c move reduce collapse to rangeify (#12845) 2025-10-23 15:44:17 +02:00
Sieds LyklesandGitHub 04b3e51f1b remove old reduce collapse rule (#12889) 
* comment this out

* remove
 2025-10-23 13:51:49 +02:00
George HotzandGitHub 92778c7a8b rename opts to ren, add store ranges back (#12856) 
* rename opts to ren

* fix docs and bring store back
 2025-10-22 09:15:38 +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
Sieds LyklesandGitHub 1df9c7d7e7 reduce_collapse uses symbolic_flat (#12766) 
* sym->symbolic_flat

* cast invalid drops invalid
 2025-10-19 12:27:47 +02:00
Sieds LyklesandGitHub 79c2f1ae26 remove reduce_rangless and replace with reduce_unparented (#12749) 2025-10-17 04:46:05 +02:00
Sieds LyklesandGitHub 55db1b0e0e reduce where that is cut from two sides (#12733) 
* better rule

* correct pattern

* shorten line
 2025-10-16 15:25:15 +02:00
Sieds LyklesandGitHub 8f740e07ff no broadcasting/vectors in reduce collapse (#12729) 2025-10-16 13:22:57 +02:00
Sieds LyklesandGitHub 99aa3bd5f9 reduce collapse reduce only the cut range (#12687) 2025-10-15 13:57:41 +02:00