nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity
13,834 Commits 334 Branches 15 Tags
move_mop_cleanup
Commit Graph
372 Commits
Author SHA1 Message Date
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
George HotzandGitHub 0035bb6fa8 only front reduce (#16901) 
* only front reduce

* lint + spec

* that should fail

* meh
 2026-07-07 09:55:07 -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
George HotzandGitHub 40d112d4d6 remove ptr=True from index (#16898) 2026-07-06 18:34:18 -07:00
George HotzandGitHub 6b8b2f5aeb remove PtrDtype from tests (#16896) 
* remove PtrDtype from tests

* fixes

* not needed

* no -1
 2026-07-06 18:22:22 -07:00
George HotzandGitHub 9512dc30f4 remove vec from const (#16889) 
* remove vec from const

* reject in spec

* remove all vector dtypes

* not needed

* remove that

* clean up invalid
 2026-07-06 12:23:07 -07:00
George HotzandGitHub c92a4c0442 move devectorizer remains into coalese (#16869) 
* move devectorizer remains into coalese

* fix flaky
 2026-07-04 15:36:45 -07:00
George HotzandGitHub a145fdce4b remove the .gep method (#16867) 2026-07-04 14:07:09 -07:00
George HotzandGitHub 3fd6f3d28c no more GEP (#16866) 2026-07-04 13:42:28 -07:00
George HotzandGitHub d76bcbaa02 deletions after new codegen (#16864) 
* deletions after new codegen

* delete the expander
 2026-07-04 13:12:11 -07:00
George HotzandGitHub c7e7687bd3 new codegen, try 3 (#16781) 
* new codegen, try 3 [pr]

* reduce

* diffs from cg2

* minor fixes

* mergable

* new local buffers

* fixes

* correct fix

* add barrier, and image is after that

* fix get

* wmma work

* devec wmma

* fixes for wmma

* slightly more flexible

* fix tensor cores

* fix linter

* tests

* remove invalid?

* remove extra

* wmma merge

* flip reshape/permute on wmma

* broadcast_binary None

* update symbolic for vecless

* x86 fix

* fix custom llama kernel

* is_ptr hack

* fix pre-commit

* fix wmma

* simpler

* fixes

* better hreduce

* move reduce axes

* all wmma tests pass

* allow stacked wmma

* permute cleanups

* late permute

* correct permute flip

* fix group for reduce

* fix wmma permuted

* all pass

* mop cleanup

* new test

* push permute/reshape

* revert that

* half
 2026-07-04 12:55:44 -07:00
chenyuandGitHub 709251e22d more generic fold_add_divmod_recombine [pr] (#16854) 2026-07-03 22:00:59 -04:00
George HotzandGitHub 41d6731bfd reduce removes ones (#16847) 
* reduce removes ones

* test changes

* lil clean
 2026-07-03 13:10:08 -07:00
chenyuandGitHub 6345c2883d simpler get_grouped_dims (#16842) 
works for more than 3D, also deleted get_contraction
 2026-07-03 10:29:11 -04:00
geohot 0240998b06 hotfix: that test_ellipsis_with_int test was important 2026-07-02 19:19:38 -07:00
George HotzandGitHub e6fbede157 changes from new codegen (#16836) 2026-07-02 17:01:37 -07:00
chenyuandGitHub 2e62dd308d move fs_load and fs_store to nn (#16821) 
similar to safe_load and safe_save
 2026-07-01 22:01:15 -04:00
chenyuandGitHub 4b3de041e2 delete Ops.DEVICE (#16820) 
85 ops
 2026-07-01 12:20:05 -04:00
chenyuandGitHub 372774fc41 empty to CreationMixin [PR] (#16811) 2026-06-30 20:42:42 -04: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
geohot af0e339114 test fix 2026-06-29 18:37:49 -07: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
chenyuandGitHub e836899336 remove Tensor conv2d and dot override [PR] (#16792) 2026-06-29 09:58:21 -04:00
nimlgenandGitHub d9c65bd843 viz: fix walk rewrite (#16791) 2026-06-29 16:32:32 +03:00
George HotzandGitHub afe9fcaec8 remove dtype.vec from transcendental (glm) [pr] (#16773) 
* remove dtype.vec from transcendental (glm) [pr]

* more cleanups

* fix fuzzer
 2026-06-26 21:41:31 -07:00
George HotzandGitHub 59aec35a7d more devec cleanups (glm) (#16772) 2026-06-26 19:27:04 -07:00
George HotzandGitHub a6ed4237e5 remove Ops.VCAT and Ops.PTRCAT (glm) (#16771) 
* remove Ops.VCAT and Ops.PTRCAT (glm)

* fix stack, and remove useless casts
 2026-06-26 18:35:38 -07:00
chenyuandGitHub 5d1867a6c3 more Tensor.training -> TRAINING [PR] (#16770) 
code and test and doc
 2026-06-26 20:11:59 -04:00
chenyuandGitHub 434e233640 LUNIQUE -> ParamArg [PR] (#16756) 2026-06-26 09:04:49 -04:00
George HotzandGitHub 5fcb21c1b3 remove vectorize/const symbolic rule [pr] (#16759) 
* remove vectorize/const symbolic rule

* post index symbolic is late

* simpler

* less gates

* that's fine

* this

* work

* don't rerun

* this order

* late removal of index dtype

* skip that test

* it's fine with extra symbolic
 2026-06-25 17:05:14 -07:00
chenyuandGitHub d953eb95d5 don't use Ops.UNIQUE in Buffer [PR] (#16754) 
build new_buffer with ParamArg
 2026-06-25 16:00:59 -04:00
George HotzandGitHub e72ecd101f split decomp into files (#16745) 
* split decomp into files

* that's in op
 2026-06-25 08:40:02 -07:00
George HotzandGitHub 54887cbbdf move decomp files into codegen (#16744) 2026-06-25 08:19:02 -07: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 0a8e61d0c5 switch to the new memory coaleser [pr] (#16716) 
* switch to the new memory coalese

* move that stuff

* copy in allowed length logic

* mulitple buffers

* new coalese is better

* fine

* earlier

* fixes

* work

* work

* valid

* stack on index const
 2026-06-23 18:03:48 -07:00
chenyuandGitHub ce87d80911 better _drop_valid_stmts [pr] (#16719) 
also dropped the unused is_increasing
 2026-06-23 19:35:01 -04:00
chenyuandGitHub 33b635d23a Tensor.train -> TRAINING [PR] (#16705) 
* Tensor.train -> TRAINING [PR]

* doc
 2026-06-22 15:13:22 -04:00
chenyuandGitHub 2d8b802958 contiguous in wino conv (#16696) 
also fixed test_counters
 2026-06-21 17:11:46 -04:00
chenyuandGitHub ba1d3baae8 masked_select and nonzero to mixin [PR] (#16695) 
with a .data stub
 2026-06-21 15:10:44 -04:00
chenyuandGitHub 58ff75272e const_like and invalids to mixin [PR] (#16690) 
* const_like and invalids to mixin [PR]

* empty_like

* einsum

* type
 2026-06-21 00:02:29 -04:00
George HotzandGitHub 925c49ce99 use placeholder in tests (#16672) 2026-06-18 20:51:44 -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
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
George HotzandGitHub 53a1226a49 STACK 0 is dtype void (#16650) 
* STACK 0 is dtype void

* spec for stack

* fix gemm group + END shape

* bump
 2026-06-17 16:28:32 -07:00
George HotzandGitHub aef85ddc4d addrspace special/range (#16647) 
* addrspace special/range

* just include indexing

* define var is alu

* bring old ignore indexing back

* mults to fix

* fixes

* ALU

* fixes
 2026-06-17 15:57:37 -07:00
George HotzandGitHub d631716858 remove const without STACK (#16639) 
* remove const without STACK

* fix GEP rewrite

* fix null tests

* fix openpilot regression

* it's 10 in CI
 2026-06-16 21:25:42 -07:00