sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions Activity
13,260 Commits 336 Branches 15 Tags
fix_src_spec
Commit Graph
373 Commits
Author SHA1 Message Date
George HotzandGitHub 8294d105a7 Update the spec in spec.py to match the current state (#16132) 
* start work on specv2

* more spec

* more spec

* fix amd emulator

* more spec

* more

* fix test_uop_graph

* move those

* spec=2

* skip those questionable tests

* ptx fix

* more spec=2

* store

* allow custom function in tensor

* spec 2

* fix beam search for tensor cores

* delete the old specs

* fix import
 2026-05-11 20:07:47 -07:00
George HotzandGitHub daed602569 rename BUFFERIZE to STAGE (#16125) 2026-05-10 09:26:46 -07:00
chenyuandGitHub b2a682ec60 remove _shape check in pm_mops [pr] (#16120) 
seems fine now
 2026-05-09 17:54:22 -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 d651b4bbf0 SPEC=3 checks the shape (#16001) 
* SPEC=3 checks the shape

* buffer view

* Revert "buffer view"

This reverts commit ffd87889a9.

* buffer view hack

* fix ptx
 2026-04-30 18:41:37 -07:00
George HotzandGitHub 46a36a838a small dtype shapes fixups (#15984) 2026-04-29 19:40:38 -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
nimlgenandGitHub 96165ff0d1 validate_with_cpu as rewrite (#15938) 
* validate_with_cpu as rewrite

* compil

* x

* linter

* moved

* fix
 2026-04-26 19:58:53 +03:00
nimlgenandGitHub e0ff6cc15c remove old schedule (#15930) 
* remove old schedule

* tests

* r

* x
 2026-04-25 16:46:36 +03:00
George HotzandGitHub 2d7fa58e61 fix shapes to match vecless (#15866) 
* fix shapes

* need to simplify shapes
 2026-04-22 18:27:46 +08:00
chenyuandGitHub d08b5d0a3b full to mixin (#15840) 
with unique_const
 2026-04-21 10:53:43 -04:00
nimlgenandGitHub ae9b84d32f rm beam uop (#15844) 2026-04-21 13:10:26 +03:00
chenyuandGitHub 72ecc61ca8 use more UOp method [pr] (#15821) 
instead of constructing UOp directly
 2026-04-20 09:17:56 -04:00
nimlgenandGitHub 23ca680a3a run_linear (#15784) 
* run_linear try 2

* x

* f

* tests

* ctx, cleaner

* r

* x
 2026-04-17 22:44:16 +03:00
chenyuandGitHub f0c12a2004 another form of assign to itself (#15770) 2026-04-16 15:17:19 -04:00
George HotzandGitHub d1cce7a476 put the ranges on store instead of after (#15759) 
* put the ranges on store instead of after

* better assert

* fix stuff

* comment out slow rules i don't understand

* simpler rule

* closer

* return false for store

* fix loop

* only a few schedule failures remain

* remove stores to self

* all tests pass locally

* remove junk

* regression test and fix

* better test, bump broken torch count

* bugfix with regression test

* new fusion is better
 2026-04-16 19:06:40 +08:00
George HotzandGitHub d24466c844 CALL with return value is FUNCTION (#15758) 
* CALL with return value is FUNCTION (GPT try)

* cleanups
 2026-04-16 13:25:07 +08:00
chenyuandGitHub 218d6b8988 delete old UOp.size [pr] (#15756) 2026-04-15 23:21:00 -04:00
George HotzandGitHub 1ae6528bb6 move schedule into schedule (#15736) 
* move schedule into schedule

* callify to root

* sched docs
 2026-04-15 11:03:25 +08:00
wozeparrotandGitHub 2b8d303f75 allreduce in precast dtype (#15689) 2026-04-13 20:24:12 -07:00
George HotzandGitHub 7610bdc59e block multistore, it's not supported (#15708) 2026-04-13 20:57:59 +08:00
George HotzandGitHub 4c1fb18a09 Revert "Revert "Tests for GatedDeltaNetBlock + fix multi after assign issue (…" (#15703) 
This reverts commit 0cec42db71.
 2026-04-13 19:09:38 +08:00
George HotzandGitHub 0cec42db71 Revert "Tests for GatedDeltaNetBlock + fix multi after assign issue (#15700)" (#15702) 
This reverts commit 6f5d756282.
 2026-04-13 19:06:44 +08:00
George HotzandGitHub 6f5d756282 Tests for GatedDeltaNetBlock + fix multi after assign issue (#15700) 
* broken after/assign test

* test for GatedDeltaNet

* better comments

* fix issue 1 with multi kernel

* fix 2

* fix

* linter

* public api + cleanup
 2026-04-13 18:43:23 +08:00
chenyuandGitHub 0254cfe642 move usum and uprod to mixin (#15690) 
and used it to clean up ops and tensor
 2026-04-12 11:42:24 -04:00
chenyuandGitHub f0eaac4235 reduce mixin (#15523) 2026-03-30 05:23:58 -04:00
chenyuandGitHub f485d0b664 UOp.sum -> usum, prod -> uprod [pr] (#15522) 
rename to prep reduce mixin
 2026-03-29 04:51:55 -04:00
chenyuandGitHub bcc08307da removed unused named arg in rules [pr] (#15414) 2026-03-22 09:25:46 -04:00
George HotzandGitHub c13d9d29ff add SHAPED_WMMA (#15400) 
* add SHAPED_WMMA

* shaped wmma

* less bad
 2026-03-21 16:16:03 +08:00
George HotzandGitHub 41a9b09683 minimal vec in amd_copy_matmul (#15398) 
* minimal vec in amd_copy_matmul

* unified

* unify

* reshape/permute

* cleanups

* simpler

* move index

* cleanups

* more shared
 2026-03-21 14:57:21 +08:00
sirhcmandGitHub 1560b534a5 remove IMAGE=2 (#15312) 2026-03-20 06:26:52 -04:00
George HotzandGitHub 3b75d8a7a2 fix double after bug in rangeify (#15381) 2026-03-20 14:53:46 +08:00
chenyuandGitHub da1700e16b dtypes.index -> dtypes.weakint (#15377) 2026-03-20 01:08:46 -04:00
George HotzandGitHub 5524916e39 llama compute gradients explicitly + 243 GB of RAM on MP=8 (#15343) 
* llama compute gradients explicitly

* apply grads

* fix multi issue

* multi BUFFER_VIEW support

* simpler

* skip the flaky test
 2026-03-18 19:54:40 +08:00
George HotzandGitHub 6109117af1 anonymous buffers are Invalid (#15336) 
* anonymous buffers are Invalid

* unique_const

* work

* remove invalid writes

* test_anonymous_buffers_in_function
 2026-03-18 14:52:56 +08:00
chenyuandGitHub 1283b57b4e update fix_store_after_hazard (#15309) 
actual gate is just not CONTIGUOUS, also don't need to check against full backward_slice
 2026-03-16 23:55:59 -04:00
George HotzandGitHub 3ff03be413 call always has tuple (#15297) 
* call always has tuple

* fix pre-commit and simplify

* update

* fix

* move that assert

* tuple

* fix multi

* cleanups

* fix merge
 2026-03-17 10:58:46 +08:00
chenyuandGitHub 02afb45f29 remove UOp.assign [pr] (#15300) 
* remove UOp.assign [pr]

it's all store and after, UOp is immutable

* fix test
 2026-03-16 21:45:41 -04:00
chenyuandGitHub 20799df10b remove Ops.ASSIGN [pr] (#15294) 
goodbye
 2026-03-16 05:22:21 -04:00
chenyuandGitHub a0d1444790 Tensor.assign is store+after [pr] (#15288) 
* Tensor.assign is store+after [pr]

* put that back
 2026-03-16 04:04:55 -04:00
George HotzandGitHub 08662bc4ab add TUPLE/GETTUPLE, simple tests pass (#15286) 
* simple tuple stuff passes

* resolved
 2026-03-16 15:06:02 +08:00
chenyuandGitHub cd14e8e64b allocations contiguous is store+after (#15280) 2026-03-15 11:58:40 -04:00
sirhcmandGitHub d753c5d7e5 IMAGE=1 image_conv2d pads for bank conflicts (#15252) 2026-03-14 07:59:16 -04:00
chenyuandGitHub 13eec8fbe8 remove unused assign rules [pr] (#15268) 2026-03-14 05:37:49 -04:00
George HotzandGitHub 06d7cddb33 amd_copy_matmul is cleaner (#15248) 
* amd_copy_matmul is cleaner

* it runs

* replicated stuff

* add tid there

* it runs

* cleanup

* x.src[1]

* flatten

* move that

* keep that assert
 2026-03-14 12:56:09 +08:00
chenyuandGitHub b3600e4774 don't emit assign in transform_precompiled_call [pr] (#15262) 2026-03-13 22:42:35 -04:00
chenyuandGitHub 576e7f985f remove handle_assign_mops [pr] (#15249) 2026-03-13 01:53:21 -04:00
George HotzandGitHub bdd62fd484 remove unneeded realize map entries (#15229) 
* remove unneeded realize map entries

* not that
 2026-03-12 14:23:19 +08:00
chenyuandGitHub 39b0f4bcc1 remove Ops.THREEFRY in remove_bufferize [pr] (#15220) 2026-03-11 05:30:33 -04:00
chenyuandGitHub 6489a6f212 Revert "remove mop_cleanup [pr] (#15217)" (#15218) 
This reverts commit 6b50df940a.
 2026-03-11 04:17:56 -04:00