nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity
13,747 Commits 334 Branches 15 Tags
new_devec
Commit Graph
49 Commits
Author SHA1 Message Date
chenyuandGitHub 3a9b87e48f remove Ops.DEVICE from Ops.PROGRAM [PR] (#16797) 2026-06-29 17:41:31 -04:00
chenyuandGitHub 33b635d23a Tensor.train -> TRAINING [PR] (#16705) 
* Tensor.train -> TRAINING [PR]

* doc
 2026-06-22 15:13:22 -04:00
sirhcmandGitHub 4d893f626a move a bunch of test_schedule to null (#16578) 2026-06-11 20:26:34 -04:00
chenyuandGitHub 7e7b481ba7 less CONST(DEVICE) (#16452) 
* less CONST(DEVICE)

no DEVICE for single device in const_like, multi has other issues

* maybe

* that?
 2026-06-01 15:55:12 -04:00
chenyuandGitHub b7280705a7 limit CONST(UNIQUE) to invalids only (#16432) 2026-05-29 16:02:06 -04:00
chenyuandGitHub c33b767407 bring back test and torch backend change for unique const (#16403) 2026-05-27 15:16:08 -04:00
chenyuandGitHub 945ed4f689 revert const unique changes (#16395) 2026-05-27 00:06:41 -04:00
George HotzandGitHub 156a4438d9 rename BUFFER_VIEW to SLICE (#16391) 
* rename BUFFER_VIEW to SLICE

* fix comments
 2026-05-26 18:15:00 -07:00
chenyuandGitHub d861c50dce remove unique_const (#16382) 2026-05-26 13:53:31 -04:00
chenyuandGitHub 9b00defc8c Revert "remove unique_const (#16372)" (#16380) 
This reverts commit 09019d6761.
 2026-05-26 12:30:07 -04:00
chenyuandGitHub 09019d6761 remove unique_const (#16372) 
* remove unique_const

* fix SDWA thing

* that?
 2026-05-26 12:18:03 -04:00
chenyuandGitHub 5d5e02871f remove Tensor.from_uop (#16344) 
and no device for const in Tensor init
 2026-05-24 18:53:09 -04:00
chenyuandGitHub e694d7f222 more deviceless const prerequisites [pr] (#16256) 
* more deviceless const prerequisites [pr]

* remove that

* arange.contiguous -> arange.clone in tests

arange will become deviceless const soon, update tests where it needs to be a buffer
 2026-05-18 23:14:12 -04:00
chenyuandGitHub 8631b6f17d remove use of requires_grad in test/ (#16237) 2026-05-16 17:21:07 -04:00
wozeparrotandGitHub d11f4d0ec2 fix: don't copy on slice of DP weight (#16089) 2026-05-07 17:58:01 -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 4164666c72 programinfo (#15942) 
* programinfo

* fix

* m

* x

* x

* changes

* x

* fix

* rm
 2026-04-27 23:12:03 +03:00
nimlgenandGitHub bb652352c7 remove execitem (#15932) 
* remove execitem

* f

* x
 2026-04-25 19:33:04 +03:00
nimlgenandGitHub a5e9ea7a60 remove schedule batch 4 (#15927) 
* remove schedule batch 4

* fini
 2026-04-25 12:36:55 +03:00
nimlgenandGitHub d3378010ee schedule() -> schedule_linear() in tests (batch 1) (#15915) 
* schedule_with_vars -> linear_with_vars in tests

* tests batch 1

* batch 2

* estimate_uop

* simpler

* rm
 2026-04-24 23:40:53 +03:00
nimlgenandGitHub f2751955cb remove linear_to_schedule from tests (#15912) 
* remove linear_to_schedule from tests

* x
 2026-04-24 20:02:10 +03:00
nimlgenandGitHub 01ac1c8c15 remove all run_schedule from tests (#15846) 2026-04-21 12:02:10 +03: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 8bd4fead26 UOp.size -> prod(max_shape) (#15755) 
and more test updates
 2026-04-15 22:41:30 -04:00
chenyuandGitHub 10c262ced8 update tests that use UOp.size (#15753) 2026-04-15 21:58:27 -04: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 4ac2552642 improve ReduceMixin.all (#15544) 
use prod instead of min since `mul` lowered to `and` directly
 2026-03-31 07:54:27 -04:00
chenyuandGitHub 89ec22131a tests to show double negation in min is not cancelled (#15543) 2026-03-31 06:59:13 -04:00
chenyuandGitHub 7c8f992894 move EXPAND dtype cast back to gradient.py (#15481) 
only a concern for gradient, not mixin
 2026-03-25 19:25:26 -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
sirhcmandGitHub ffaafd391a Invalid in Tensor (#15154) 2026-03-10 02:49:54 -04:00
sirhcmandGitHub 592f9bf6c6 set OPENPILOT_HACKS=1 to enable replace assign (#15123) 2026-03-04 05:26:04 -05:00
George HotzandGitHub d483e4153a buffer view is like buffer (#15082) 
* buffer view is like buffer

* fix

* swap_reshape_shrink

* contiguous on gguf, fix overlap

* revert that

* _device_supports_view

* this

* fix that test

* 0 buffers

* that test was wrong

* this

* check correct size

* contig BUFFER_VIEW

* this

* fix tests

* buffer view tests

* om

* fix torch

* no MOCKGPU

* skip
 2026-03-03 09:52:33 +08:00
chenyuandGitHub 14d1c5fdfd assign fusion tests on detach and contiguous_backward (#15092) 2026-03-02 15:21:51 -05:00
chenyuandGitHub d345f7f5dc remove _pending_assigns (#15040) 2026-02-26 22:38:10 -05:00
qazalandGitHub c5029fa460 jit case with Tensor.empty input, realized means allocated (#14930) 
* simple failing jit test case with Tensor.empty

* this used to exist in ops.py...

* Revert "removed if self.buffer.is_allocated() in realized (#14836)"

This reverts commit 72cf603805.
 2026-02-21 16:33:55 +09:00
George HotzandGitHub df7774661a remove late numbering of UOps (#14923) 
* remove late numbering of UOps

* stupid fix

* dead code
 2026-02-21 09:18:48 +08:00
George HotzandGitHub 55d3a5def9 preallocate all realized buffers (#14823) 
* preallocate all realized buffers

* contiguous

* work

* comment that out

* move to schedule

* better

* correct fix

* just buffer

* disk bufs

* fixes disk tensor stuff

* fix symbolic stuff

* fix multi

* 162 failures

* bugfixes

* don't check that anymore

* fix schedule tests

* mnist should be contiguious

* type and buffer

* fix tests

* shrink axis correction

* mypy fixes

* tests skips

* same 37 failures

* dedup

* no shrink in the graph

* 29 failures

* skips

* fix custom kernel

* fix training

* those optimizations aren't supported currently

* simpler

* more correct

* tests

* 14 failures

* works

* fix that test

* broken

* 11 failures

* only kernel counts left

* fixes

* all tests pass

* remove tensor_map

* op test

* 200 -> 230

* test fixes

* fixes

* revert test_tiny thing

* guard

* revert that

* test tiny passes

* no contigs there

* base realize back

* Revert "no contigs there"

This reverts commit c45bb9fcfd.

* revert that

* chop many assigns

* 12 failures

* fix tests

* tests

* apply after

* pre-commit

* remove old code

* delete that

* fix types

* remove extra contig

* fix dataloader

* torch fix

* disk fix

* update kernel fusion numbres

* runs on amd

* restore kernel count

* add that rule back

* that

* disable that

* wrong

* add the correct rule for that folding

* more tests

* guard c1.arg

* no newlines

* realize those

* split into a different file

* remove detach/contig back

* skip 2

* update that
 2026-02-20 20:05:54 +08:00
George HotzandGitHub fc5677c28b resnet dataloader + more test cleanups (#14899) 
* resnet dataloader

* tests
 2026-02-20 10:05:47 +08:00
George HotzandGitHub 2f0f8b5776 more test relaxations from prealloc_bufs (#14880) 2026-02-19 14:23:28 +08:00
George HotzandGitHub ab61c16730 fixes and test relaxations from prealloc_bufs (#14875) 
* fixes and test relaxations from prealloc_bufs

* fix error type and guard _mop

* revert that

* contiguous makes extra/torch_backend/test_kernel_fusion.py fail
 2026-02-19 11:37:25 +08:00
chenyuandGitHub f771de6738 gc.collect() to get the correct GlobalCounters.mem_used in tests (#14868) 
test can be flaky if gc happens in between
 2026-02-18 15:01:23 -05:00
George HotzandGitHub ab55e8c6b9 assign should be used as output buffer (#14845) 
* assign should be used as buffer

* late removed

* the fix

* better fix

* backward slice
 2026-02-18 09:37:46 +08:00
chenyuandGitHub 72cf603805 removed if self.buffer.is_allocated() in realized (#14836) 
automatically fixes is_realized issue for empty
 2026-02-17 15:35:56 -05:00
George HotzandGitHub 8dc46dde07 everything has dtype.long now (#14661) 
* everything has dtype.long now

* int64/uint64 are everywhere now

* that doesn't work
 2026-02-10 15:08:50 +08:00
George HotzandGitHub cc9bf8ccbc move more to null/unit tests (#14658) 
* move more to null tests

* move test_gc

* no test fusion op
 2026-02-10 13:35:17 +08:00
George HotzandGitHub d59e6e7a37 move more tests to test/null, split some existing ones (#14512) 
* move more tests to test/null, split some existing ones

* null work

* null work

* move more

* fixes

* move PIL

* PIL in CLIP

* don't move that
 2026-02-03 20:20:20 +08:00