mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity
13,633 commits 280 branches 15 tags 325 MiB
Commit graph

112 commits

Author SHA1 Message Date
George Hotz
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 Hotz
7e6d617935
 		addrspace cleanups (#16565) 
* addrspace cleanups

* bumps

* eh, relax a little
 2026-06-10 15:57:18 -07:00
chenyu
2c915c61ed
 		no CONST(DEVICE) in torch_backend (#16499) 2026-06-04 00:26:47 -04:00
chenyu
8a4203638a
 		make full with buffer=False deviceless (#16483) 
affects arange and eye
 2026-06-03 12:35:59 -04:00
chenyu
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
Christopher Milan
434cfa96a3
 		ci: no fetch in backend tests (#16438) 
should make for less actions cache thrashing
 2026-05-29 17:11:16 -04:00
George Hotz
edca5df25a
 		flip offset and shape in pad and shrink (#16414) 
* flip offset and shape in pad and shrink

* dumb test
 2026-05-28 11:58:19 -07:00
George Hotz
8ee3a37524
 		shrink/pad use (new_shape, offset) (#16405) 
* shrink uses offset and shape

* pad does too

* fix
 2026-05-27 15:13:08 -07:00
chenyu
c33b767407
 		bring back test and torch backend change for unique const (#16403) 2026-05-27 15:16:08 -04:00
chenyu
945ed4f689
 		revert const unique changes (#16395) 2026-05-27 00:06:41 -04:00
chenyu
d861c50dce
 		remove unique_const (#16382) 2026-05-26 13:53:31 -04:00
chenyu
9b00defc8c
 		Revert "remove unique_const (#16372)" (#16380) 
This reverts commit 09019d6761.
 2026-05-26 12:30:07 -04:00
chenyu
09019d6761
 		remove unique_const (#16372) 
* remove unique_const

* fix SDWA thing

* that?
 2026-05-26 12:18:03 -04:00
chenyu
31424cda71
 		Tensor.requires_grad -> is_param (#16325) 
for optimizer
 2026-05-21 19:39:57 -04:00
chenyu
409bb0c9ad
 		requires_grad cannot be None (#16212) 
final goal is to remove requires_grad, first change the default to True, and don't allow None
 2026-05-15 02:01:04 -04:00
chenyu
3942a80f66
 		fix wrong kwargs passed into rands (#16149) 
working towards explicit args for these
 2026-05-11 22:22:06 -04:00
Vikram Rangarajan
effa263865
 		Torch backend aten::cat.out fix (#16121) 
* Handle empty 1D tensors in cat_out

* Undid other changes

* Fixed torch cat

* Improved cat.out, added more tests

* Cleaned code

* Type hinted dim

* Removed whitespace
 2026-05-11 16:28:16 -07:00
George Hotz
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
chenyu
1483f7e71c
 		support shift by Tensor (#15623) 
* support shift by Tensor

* use mixin
 2026-04-06 15:14:57 -04:00
chenyu
6e30a5f5ea
 		update shifts in torch backend (#15622) 2026-04-06 14:08:33 -04:00
wozeparrot
b45edeb965
 		fix: rand supports large tensors (#15329) 2026-03-17 15:45:41 -07:00
chenyu
842c978df3
 		remove staticmethod dtypes.max/min (#15227) 
always use x.dtype.max/min
 2026-03-11 23:11:24 -04:00
Roelof van Dijk
d65923bda5
 		tensor.py: add normalize function (#15159) 
* tensor.py: add normalize function

* p==0 should match torch
 2026-03-05 18:55:53 +08:00
chenyu
71f228f80f
 		test exact kernel count in torch_backend/test_kernel_fusion (#15091) 2026-03-02 14:26:32 -05:00
George Hotz
8ef5544e4a
 		realized PYTHON copies (#14934) 
* realized PYTHON copies

* comment that out

* fix that test

* append afters

* contig

* disk copies

* should be 124

* 332
 2026-02-21 20:29:31 +08:00
George Hotz
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
chenyu
9052db678f
 		remove allow_shape_mismatch in Tensor.replace (#14536) 
move all logic to torch_backend and not hacking Tensor method
 2026-02-04 12:38:18 -05:00
chenyu
67f91e897b
 		UOp.is_contiguous -> UOp.has_buffer_identity [pr] (#14530) 
one more confusing buffer related method, but it's definitely not is_contiguous
 2026-02-04 09:21:26 -05:00
chenyu
e3601788fa
 		update torch backend function (#14333) 
those have tensor.py implementation
 2026-01-25 16:39:34 -05:00
chenyu
986e865830
 		fix TINY_BACKEND=1 cumsum (#14138) 
* fix TINY_BACKEND=1 cumsum

old hack was wrong, need to apply contiguous on the input

* test time

* test_linalg_svd is slow
 2026-01-14 09:54:49 -05:00
chenyu
fe00682502
 		clean up svd tests (#14133) 
removed from test_ops and added to TestTorchBackend
 2026-01-13 16:32:21 -05:00
chenyu
e610821c52
 		Tensor.cummin and Tensor.nonzero (#14131) 2026-01-13 15:09:56 -05:00
chenyu
176a934ddd
 		Tensor.diagonal support offset and dims (#14130) 2026-01-13 14:49:06 -05:00
chenyu
2a217ba206
 		tinybackend isin and log10 (#14120) 
can use tinygrad directly
 2026-01-13 14:14:09 -05:00
chenyu
05fcb57696
 		also return index in Tensor.cummax (#14117) 
* also return index in Tensor.cummax

* fix
 2026-01-12 22:42:10 -05:00
Roelof van Dijk
1058748440
 		torch backend: no aten.detach for torch 2.10 compat (#13381) 
* this works, less cpp?

* simpler = better

* keep torch 2.9 working as well
 2025-11-20 09:12:15 -08:00
Roelof van Dijk
0dc2ff431d
 		fix: revive torch backend (#13280) 
* fix: revive torch backend

* as_strided view vs copy

* Revert "as_strided view vs copy"

This reverts commit 82a61223f2.

* add extra tests (move inplace, add fusion tests)

* better fusion with inplace_op

* no optimizer hooks (break mnist training fusion)

* split off fusion tests in separate file, assert on resnet fusion

fix: remove comments

* cleanup, reduce diff

* reduce diff

* better fusion and identity checks

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2025-11-19 15:26:50 -08:00
Daniel
d65bd669f8
 		update tiny torch backend hook (#12575) 
* update the backend to fix torch deprecation warning

* use param_hook to avoid full backward hook needlessly firing on inputs which do not require gradients

* fix indentation

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-10-15 14:02:33 -04:00
George Hotz
fb61f3519f
 		remove assign contiguous hack (#12659) 
* remove assign contiguous hack

* remove bad contiguous usage in torch backend

* assign
 2025-10-14 16:42:14 +08:00
chenyu
e701106a64
 		remove FUSE_ARANGE (#12511) 
it was the default already
 2025-10-08 04:54:07 -04:00
George Hotz
0f25b4b289
 		move frontend dir to nn [pr] (#12470) 2025-10-07 10:42:22 +08:00
chenyu
12a910f1d2
 		update torch 2.8 (#12172) 
support _reshape_alias. something is wrong with one case of unfold
 2025-09-14 15:19:03 -04:00
chenyu
fb8ee02424
 		Tensor.logaddexp (#11793) 2025-08-23 09:15:00 -04:00
Joshua Kissoon
c44760c89d
 		torch backend: fix arange, add linalg.cross, add tests (#11628) 2025-08-11 23:34:41 -04:00
kevvz
c3cfcb50cb
 		Add linalg_det and test for torch backend (#11405) 
* add linalg_det and test

* space

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-07-30 22:04:44 -04:00
वेदांत
e368628736
 		Add amin support to Tensor operations in Torch backend (#11290) 
* intiger div mod fix

* Revert "intiger div mod fix"

This reverts commit d5d2f201bf.

* feat arg_min support

* tets update

* test fix
 2025-07-21 09:14:08 -04:00
kevvz
b7af9cf849
 		clean svd tests, set full_matrices false in torch backend (#11113) 
* clean tests, set full_matrices false

* add more shape asserts
 2025-07-06 13:55:49 -04:00
chenyu
ba88ec3ad0
 		pipe linalg svd to torch (#11109) 
and found a bug in svd
 2025-07-06 08:37:25 -04:00
chenyu
49bba2f0a0
 		improve test_nll_loss (#10986) 
build target and weight tensors outside so it tests backward too.
 2025-06-26 02:46:55 -04:00
chenyu
ffb032e31d
 		test_diagonal touchup (#10962) 2025-06-24 15:51:19 -04:00