sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 8 Activity
14,536 Commits 336 Branches 15 Tags
master
Commit Graph
58 Commits
Author SHA1 Message Date
nimlgenandGitHub 55c8e18930 hcq2 spec2 (#17798) 2026-08-28 12:51:55 +03:00
chenyuandGitHub 4bdc865131 delete unused rewrite rules [PR] (#17748) 2026-08-26 10:50:42 -04:00
George HotzandGitHub 9aa9e11301 compile kernels in parallel (#17629) 
* compile kernels in parallel (slop)

* cleanups

* fixes

* hand cleanups

* no PARALLEL with VIZ

* ugh, we need to remove the n from names

* no hcq2 in that test

* main for abstractions3

* fix compile server to be fork safe

* fix num cpu threads in context

* just use a lock

* fix __main__ on spawn

* compileonce

* python3 is double

* xdist sizing

* parallel compile

* fetch supports extract

* fix issues

* revert compiler server to master, drop compileonce

* normal workflow

* PARALLEL=0 for QCOMCL compiletest
 2026-08-20 16:41:45 -07:00
George HotzandGitHub 1707dca3b4 remove kernel_cnt, names are no longer unique (#17633) 
* remove kernel_cnt, names are no longer unique

* check uops, not names

* fix hcq2
 2026-08-20 13:28:25 -07:00
George HotzandGitHub c31038ff37 use KernelCountException when kernel count is being compared (#17584) 2026-08-18 16:06:03 -07:00
nimlgenandGitHub 6b35220622 cpu hcq2 (#17503) 
* cpu hcq2

* temp

* slop

* test with backpressure

* x

* x

* x

* x

* x

* x

* Dx

* save reverts

* um?

* x

* x

* call from py

* x?

* x

* submitters gone

* x

* x

* z

* Dx

* Dx

* x

* x

* fixes

* repl

* x

* f

* for now keep hcqbuffer
 2026-08-14 15:06:03 +03:00
George HotzandGitHub e1f42681fa add new schedule tests + format better (#17402) 
* add new schedule tests + format better

* assert_kernel_count
 2026-08-04 18:46:38 -07:00
George HotzandGitHub 6122b3c98f use check_schedule in tests where possible (#17400) 2026-08-04 18:17:29 -07:00
qazalandGitHub 161783d8f7 add _device_num back to ast.variables (kimi) (#17327) 2026-08-01 14:56:50 +09:00
qazalandGitHub 6d2700f0b7 failing test for unbound _device_num err in BEAM (#17326) 
* min failing test

* switch to cpu

* err
 2026-07-31 20:24:44 +09:00
George HotzandGitHub b30c7e00d4 support 2d on UNSHARD (kimi) (#17285) 
* support 2d on UNSHARD

* fixes

* Fix test and spec

* single barrier

* 2d sharding works for devices too

* cleanups

* no _rewrap
 2026-07-29 12:01:59 -07:00
George HotzandGitHub 1757067fa6 add device range as src[1] to multi (kimi) (#17264) 
* add device range as src[1] to multi (kimi)

* cleanup

* fix tests
 2026-07-28 16:39:34 -07:00
chenyuandGitHub 2983987321 clean up some tests (#17154) 
duplicated or typo or slow
 2026-07-23 11:22:18 -04:00
sirhcmandGitHub a9fbc7db7b expect _offset support, CL and WEBGPU are outliers (#17014) 2026-07-13 18:53:32 -04:00
chenyuandGitHub 9d47014fd8 first class STACK [PR] (#16986) 2026-07-11 13:55:32 -04:00
sirhcmandGitHub 8c0ba1da5c cleanup more from test/backend (#16521) 2026-06-05 18:38:46 -04:00
wozeparrotandGitHub fd13080636 deviceless const skip axis check (#16496) 2026-06-03 19:13:20 -07: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
chenyuandGitHub fa14cde05c test update for arange and eye (#16394) 
these will need explicit clone to make a buffer
 2026-05-26 22:48:34 -04: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
wozeparrotandGitHub 76fc39ccc0 gather to single device (#16354) 2026-05-25 17:27:08 -07:00
chenyuandGitHub 720a27bed8 remove many requires_grad= args (#16321) 
* remove many requires_grad= args

* doc and example

* not cifar
 2026-05-21 18:37:11 -04:00
sirhcmandGitHub 172f9493e1 move is_dtype_supported to renderer (#16226) 2026-05-20 21:19:37 -04:00
chenyuandGitHub 890b731b1e more prerequisuite test changed for deviceless const (#16264) 2026-05-19 15:43:45 -04:00
George HotzandGitHub 3f2d401464 all tests pass with NOOPT=1 (#16257) 
* all tests pass with NOOPT=1

* fix a few more

* noopt 100% pass

* noopt 100% pass
 2026-05-18 20:39:51 -07: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 d532b4f533 multi alu with deviceless const (#16251) 2026-05-18 19:31:53 -04:00
chenyuandGitHub 8631b6f17d remove use of requires_grad in test/ (#16237) 2026-05-16 17:21:07 -04:00
chenyuandGitHub 09fd80fba6 fix randperm and _multi_like drop requires_grad (#16150) 2026-05-11 23:23:34 -04:00
nimlgenandGitHub d2ab6ea7a6 remove schedule batch 3 (#15924) 
* remove shcedule batch 3

* batch 6

* batch 7
 2026-04-25 11:53:16 +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
chenyuandGitHub 7a1adfd2aa update Tensor.allclose to return Tensor (#15904) 
matches jax
 2026-04-24 08:27:17 -04:00
nimlgenandGitHub c0f77c2e1c hcq graph to linear (#15888) 
* hcq

* f

* f

* linter
 2026-04-24 12:42:49 +03:00
nimlgenandGitHub 5cf4ad2fb6 fix resolve param (#15889) 2026-04-23 17:41:44 +03:00
nimlgenandGitHub 01ac1c8c15 remove all run_schedule from tests (#15846) 2026-04-21 12:02:10 +03:00
nimlgenandGitHub c0d7135b5f do not use jit_cache in test (#15823) 
* do not use jit_cache in test

* fix
 2026-04-20 11:45:17 +03:00
chenyuandGitHub e706f408cb suppress test warnings from numpy (#15688) 2026-04-11 22:33:20 -04:00
chenyuandGitHub 1aa04eab08 simple CreationMixin (#15567) 
start with full_like, zeros_like, ones_like
 2026-04-01 23:00:56 -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
chenyuandGitHub fceb21c315 Tensor(uop) uses device from uop (#15340) 2026-03-18 02:56:06 -04: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 151608aa90 update test_multiple_to_single_device (#15056) 
follow up to #14482, add SCACHE=0 to the test
 2026-02-27 21:44:33 -05:00
chenyuandGitHub 4424757b9a update test_sharded_memory (#14956) 
cleaned up and moved to test/null
 2026-02-22 16:56:08 -05:00
George HotzandGitHub 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
chenyuandGitHub 24286c5593 fix clone for multi (#14919) 
also update empty_like to make sure it's backed by buffers
 2026-02-20 17:21:09 -05: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 0c85b93938 support shink sharded and non-sharded axes (#14874) 
simpler to just support it
 2026-02-18 20:54:10 -05:00