sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions Activity
13,834 Commits 336 Branches 15 Tags
move_mop_cleanup
Commit Graph
281 Commits
Author SHA1 Message Date
George HotzandGitHub 0035bb6fa8 only front reduce (#16901) 
* only front reduce

* lint + spec

* that should fail

* meh
 2026-07-07 09:55:07 -07:00
Ben WaldronandGitHub 357b7544d6 Allow non-const alt values for wgsl gated load (#16885) 
* Allow non-const alt values for wgsl gated load

* Add regression test

* Make z3 happy

* Change to int to match nearby test + re-run CI

* Remove ptr
 2026-07-07 08:18:02 -07: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 4faed79216 remove placeholder from is_ptr (#16893) 
* remove placeholder from is_ptr

* simpler

* needed

* remove rewriter

* fix tests
 2026-07-06 16:55:56 -07:00
sirhcmandGitHub deb5232413 skip test_index_fused_out_of_bounds if CL (#16888) 2026-07-06 13:53:58 -04:00
chenyuandGitHub a398b678cd skip test_index_fused_out_of_bounds (#16877) 
reads OOB now
 2026-07-05 09:30:37 -04:00
George HotzandGitHub a145fdce4b remove the .gep method (#16867) 2026-07-04 14:07:09 -07:00
wozeparrotandGitHub 971800b46a support none block divisible embed size (#16865) 2026-07-04 13:46:47 -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
qazalandGitHub 07f7383d29 llama: remove unused bf16 assembly gemm (#16859) 
* only hk bf16 gemm

* rm asm gemm

* more cleanup

* half isn't supported in asm gemm anymore

* more test edits

* unused

* remove TestMagicGu

* uop gemm is still tested

* minimal diff
 2026-07-04 18:41:54 +09:00
sirhcmandGitHub 353d8f1e13 use rusticl in ci (#16852) 2026-07-03 23:58:58 -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 5fb3cfb9bc remove early THREEFRY const folding [PR] (#16850) 
it folds automatically once decomposed
 2026-07-03 15:00:35 -04:00
chenyuandGitHub 59c7874724 fix isclose with scalar other (#16813) 2026-06-30 21:56:33 -04:00
George HotzandGitHub b683ff9836 test updates + addrspace work from new codegen (#16800) 2026-06-29 19:12:10 -07:00
chenyuandGitHub 3a9b87e48f remove Ops.DEVICE from Ops.PROGRAM [PR] (#16797) 2026-06-29 17:41:31 -04:00
qazalandGitHub 905b405820 fix contiguous_view_offset for scalar index (#16789) 
* simple failing test

* scalar passes
 2026-06-29 12:50:21 +09:00
chenyuandGitHub 5d1867a6c3 more Tensor.training -> TRAINING [PR] (#16770) 
code and test and doc
 2026-06-26 20:11:59 -04:00
chenyuandGitHub 8c9e8d38a9 fix permute copy (#16766) 
flip and permute copy was wrong
 2026-06-26 11:42:50 -04: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 33b635d23a Tensor.train -> TRAINING [PR] (#16705) 
* Tensor.train -> TRAINING [PR]

* doc
 2026-06-22 15:13:22 -04:00
wozeparrotandGitHub bba611bb59 gemm: fix mxfp8 on more shapes (#16677) 2026-06-19 13:28:53 -07: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 b05bea81ce x86 cleanups (fable) [pr] (#16591) 
* x86 cleanups (fable)

* support shrink

* remove ptr dtype

* move that

* is_lane helper

* Revert "is_lane helper"

This reverts commit ea4571254d.
 2026-06-19 09:04:51 -07:00
George HotzandGitHub 925c49ce99 use placeholder in tests (#16672) 2026-06-18 20:51:44 -07:00
qazalandGitHub 924bece1d5 remove some old scheduler tests (#16660) 2026-06-18 22:15:00 +09:00
chenyuandGitHub 1e08c0a07c remove NOOP from AFTER with multiple srcs (#16646) 2026-06-17 14:35:02 -04:00
chenyuandGitHub 1acc40600d indexing an after with all fully invalid stores is invalid (#16643) 
* indexing an after with all fully invalid stores is invalid

* typing cast
 2026-06-17 11:06:36 -04:00
wozeparrotandGitHub 36f6d1b064 gemm: fix bf16 atb for mp sharding (#16637) 2026-06-16 15:58:47 -07:00
geohot b1fb39502d delete that test 2026-06-14 09:42:58 -07:00
nimlgenandGitHub c43091a464 fix missing cast in cstyle (#16608) 
* fix missing cast in cstyle

* x

* x
 2026-06-13 10:04:06 +03:00
chenyuandGitHub aa32d309db fix rangeify indexing for pad/reduce (#16599) 2026-06-12 20:26:15 -04:00
qazalandGitHub 4d34590b7d llama: less E kernels (#16517) 2026-06-12 19:49:25 +09:00
sirhcmandGitHub 4d893f626a move a bunch of test_schedule to null (#16578) 2026-06-11 20:26:34 -04:00
wozeparrotandGitHub c38d6a7e3a mxfp8 part 2 (#16561) 2026-06-10 23:36:11 -07:00
George HotzandGitHub 7e6d617935 addrspace cleanups (#16565) 
* addrspace cleanups

* bumps

* eh, relax a little
 2026-06-10 15:57:18 -07:00
qazalandGitHub 34481830f1 rangeify: fix cost function for AFTER(out, CALL) (#16559) 
* simple failing test

* fix rangeify cost function

* new ops count
 2026-06-10 17:30:50 +09:00
sirhcmandGitHub f9d88d3c3a fix race in test_quantize_onnx (#16555) 2026-06-09 18:39:48 -04:00
wozeparrotandGitHub 2bdc360606 gemm: mxfp8 hipkittens gemm (#16541) 
* gemm: mxfp8 hipkittens gemm

* feat: update hipkittens

* feat: kernel signature

* clean: just kernel

* feat: from tinygrad

* feat: test

* fix: add back utils

* clean: no diff

* clean: no diff
 2026-06-09 15:20:05 -07:00
George HotzandGitHub 0336ba8eb1 buffer param arg + dsp fixups (#16528) 2026-06-07 12:07:00 -07:00
George HotzandGitHub 2a2f81dd3d remove ANON from addrspace, refactor marg (#16523) 
* remove ANON from addrspace, refactor marg

* as_shape

* as_shape is cached
 2026-06-06 09:49:09 -07:00
sirhcmandGitHub 8c0ba1da5c cleanup more from test/backend (#16521) 2026-06-05 18:38:46 -04:00
sirhcmandGitHub 9b0f75622c many jit tests belong in unit (#16508) 2026-06-04 21:36:53 -04:00
chenyuandGitHub 4a8bf07a87 remove CONST(DEVICE) (#16506) 2026-06-04 11:29:46 -04:00
chenyuandGitHub 5fad87252d no device= into arange and eye (#16503) 2026-06-04 09:21:50 -04:00
wozeparrotandGitHub fd13080636 deviceless const skip axis check (#16496) 2026-06-03 19:13:20 -07:00
chenyuandGitHub 2f83d01ccf fix deviceless materialize device (#16493) 
symbolic arange currently does not fuse, which creates a deviceless UOp post rangeify that needs a device to bufferize
 2026-06-03 19:13:21 -04:00
chenyuandGitHub 19eb72ff60 remove use of full with buffer=False and non-None device= (#16489) 2026-06-03 16:21:24 -04:00
chenyuandGitHub 8a4203638a make full with buffer=False deviceless (#16483) 
affects arange and eye
 2026-06-03 12:35:59 -04:00