github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-17 02:48:03 -05:00
Code Issues Packages Projects Releases Wiki Activity
10,417 Commits 204 Branches 14 Tags
reprocess_node_experiments
Commit Graph

10417 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
George Hotz
dde61c3852 Revert "write tests for algebraic UPat" 
This reverts commit 6538935441.
 2025-10-05 15:11:07 +08:00
George Hotz
136aeaacd3 works 2025-10-05 15:11:05 +08:00
George Hotz
18552a3040 canon 2025-10-05 14:58:52 +08:00
George Hotz
c600446299 experiments with reprocessing node 2025-10-05 14:12:40 +08:00
George Hotz
6538935441 write tests for algebraic UPat 2025-10-05 08:21:38 +08:00
qazal
865d5796f8 add a test for untested Tensor.assign behavior (#12448) 
* add a test for untested Tensor.assign behavior

* better
 2025-10-04 12:44:56 +03:00
Sieds Lykles
e74be4a140 UOp.factor and add chain sorting (#12413) 
* add ordering

* fix some tests

* fix more tests

* shorten comment

* update test

* add rule and test

* add rule and test

* remove check

* use fold_divmod_congruence instead of simplify

* adjust tests

* shorten line

* new algo

* add test

* add function to un-nest the div

* add UOp.factor

* test UOp.factor

* uop_given_valid tries to factor simplex expression

* shorten line

* symbolic_flat is back

* change that back

* fix those new tests

* new rule for ordering

* factor multiple factors

* no symbolic_flat

* symbolic_flat to there

* move that back

* fix imports

* merge correctly

* linter happy

* add rule

* add a test

* cleanup

* revert that for now

* UOp.factor returns self instead of None

* try all_candidates

* remove or_else

* post index symbolic

* add test

* maket this closer to the original

* increase mac hlb_cifar min step time

* add some ordering tests

* cleanup

* increase pytest timeout time

* check dtype
 2025-10-04 06:05:38 +02:00
Sieds Lykles
394dc24110 post index symbolic (#12446) 
* post index symbolic

* add test
 2025-10-03 23:23:03 +02:00
chenyu
9f2b69b870 enable few tests for PTX test_dtype (#12445) 2025-10-03 08:56:30 -04:00
George Hotz
0b534f71c2 recursive substitute should be O(n) (#12444) 
* recursive substitute

* even faster

* make that a single rewrite
 2025-10-03 18:29:59 +08:00
chenyu
b087663c35 RANGEIFY test_bert uses more ran somehow (#12443) 2025-10-03 04:38:53 -04:00
chenyu
940a8d5ba9 default IGNORE_OOB=1 (#12441) 
* default IGNORE_OOB=1

z3 can get very slow with RANGEIFY, also update some kernel numbers to what it is

* add to test
 2025-10-03 04:16:19 -04:00
George Hotz
d290e77a5b pend substitutes for speed (#12440) 2025-10-03 15:49:19 +08:00
nimlgen
23d310bcc1 ptx: handle i8/u8 casts correctly (#12439) 
* ptx: handle casts correctly

* notsetp
 2025-10-03 15:34:15 +08:00
hooved
1e8945a28c Training loop for Stable Diffusion mlperf (#12315) 
* add diff

* fix edit error

* match master

* point reference to specific commit

* simplify wandb logging

* remove lr test, dehardcode device

* increase stack size limit
 2025-10-03 02:45:38 -04:00
George Hotz
c7849ac593 fix test lil model (#12437) 
* fix test lil model

* 4 not 3
 2025-10-03 02:28:37 -04:00
chenyu
0f82d92b9d use float for softmax in llm.py (#12438) 
fixed numerical issue in `CPU=1 RANGEIFY=1 python3 -m tinygrad.apps.llm`
 2025-10-03 02:27:56 -04:00
George Hotz
4c63f7e786 skip copies of reshaped buffers (#12430) 
* skip copies of reshaped buffers

* always run NOOP

* comment

* comment
 2025-10-03 13:05:47 +08:00
Sieds Lykles
0047bcc535 undo loaded comparison swap (#12436) 
* add rule

* add a test
 2025-10-03 06:57:29 +02:00
chenyu
f203d8b221 update RANGEIFY kernel count and test_masked_select (#12435) 2025-10-03 00:41:34 -04:00
wozeparrot
a6dd5a224b skip webgpu tests (#12433) 2025-10-02 21:31:07 -07:00
chenyu
bf99de7b1e update a few more tests for RANGEIFY (#12434) 2025-10-03 00:16:58 -04:00
George Hotz
9cd365c12e little changes from double gemm (#12429) 
* little changes from double gemm

* split pm_group_for_reduce

* pm_add_buffers_local

* Revert "pm_add_buffers_local"

This reverts commit 4d30a91db2.
 2025-10-03 10:31:51 +08:00
Sieds Lykles
16a65b4fd0 fix test_symbolic_gcd_div hang (#12427) 2025-10-03 04:21:16 +02:00
chenyu
2d24af888b REWRITE_STACK_LIMIT (#12426) 2025-10-02 21:51:04 -04:00
hooved
1b58ef0d60 Increase stack size limit in unified_rewrite (#12424) 
* increase stack size limit

* rerun CI due to random tqdm test fail
 2025-10-03 09:06:47 +08:00
qazal
17d36d0952 don't tag MSTACK/MSELECT on global buffers (#12423) 
* don't tag MSTACK/MSELECT

* fix
 2025-10-02 13:32:15 +03:00
chenyu
7b3912d8e4 relax atol for some tests (#12422) 2025-10-02 05:04:44 -04:00
chenyu
98163832e4 update RANGEIFY test_cast_padded (#12421) 
* update RANGEIFY test_cast_padded

* update test
 2025-10-02 04:37:35 -04:00
chenyu
37beef6de3 add null bert training test in ci (#12420) 
fails with RANGEIFY `RuntimeError: children not making progress`
 2025-10-02 04:05:19 -04:00
qazal
f21851b099 ops: n^2 .device property fix (#12419) 
* test case for a long rand chain

currently failing with RANGEIFY because device propogates too deep

* skip

* ops: n^2 .device property fix

* unskip

---------

Co-authored-by: Chen-Yu Yang <chenyu@fastmail.com>
 2025-10-02 03:28:12 -04:00
b1tg
ec177c80c2 rangeify: fix test_where_fold (llvm) (#12416) 
* rangeify: fix test_where_fold (AMD_LLVM)

* rm comment
 2025-10-02 02:57:49 -04:00
qazal
13a25b2e67 rangeify: don't shape INDEX on kernelize (#12417) 2025-10-02 09:45:37 +03:00
hooved
5d9035f5a6 Eval for Stable Diffusion mlperf (#12316) 
* add diff

* rerun ci

* refactor beam workaround, add test

* fix conflict

* linting
 2025-10-02 02:35:38 -04:00
hooved
0f804c9a83 Stable Diffusion model init for mlperf (#12314) 
* include clip pr diff

* updated unet and sd init

* dehardcode default device

* revert beam hang workaround

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-10-02 02:28:41 -04:00
George Hotz
0eee93f0c0 hotfix: disable split ranges for non rangeify 2025-10-02 13:15:24 +08:00
George Hotz
583553f467 split ranges (#12411) 
* split ranges

* simpler

* split ranges

* range str

* fix test

* oops

* faster

* no group 2

* tests

* dont_sub_ranges_for_image

* revert that
 2025-10-02 12:57:22 +08:00
qazal
6fc6b51b59 fix limit_bufs with kernelize (#12415) 2025-10-02 07:49:11 +03:00
qazal
d1c868f990 fix limit_bufs with multi (#12414) 2025-10-02 05:51:56 +03:00
qazal
2fcd55583f allow less kernels in external_test_opt (#12412) 
* allow less kernels in external_test_opt

* this was always 2
 2025-10-02 05:05:42 +03:00
qazal
8b48e19ce2 skip more multi remote tests (#12410) 2025-10-02 04:50:46 +03:00
George Hotz
3770dd9d80 annotate bufferize in viz 2025-10-02 09:20:50 +08:00
qazal
5b649616ff rangeify: detect and assert cycles (#12405) 
* rangeify: assert cycles

* rng=2

* any
 2025-10-02 03:39:43 +03:00
Sieds Lykles
9a64fc0d28 Load alt value with cast try 2 (#12407) 
* add or_casted

* add tests and fix old tests

* cast load

* move that to pm_render

* add allow_any_len to gated load patterns in renderers

* slice [:2]
 2025-10-02 00:55:29 +02:00
nimlgen
3e0e0290ce increase timeout in test_module_runs (#12408) 2025-10-01 22:01:44 +03:00
Sieds Lykles
2f8ac77c25 add allow_any_len to gated load patterns in renderers (#12406) 2025-10-01 20:35:32 +02:00
George Hotz
89bed28716 split reduceop (#12404) 
* some rangeify tests fixed

* bring split reduceop to rangeify

* fix tests
 2025-10-01 18:45:16 +08:00
George Hotz
74ee305948 some rangeify tests fixed (#12403) 2025-10-01 18:23:37 +08:00
qazal
f198a9e1ba skip test_multihost_aware_schedule, assign devices mismatch (#12396) 
* minimal failing remote test

* this should've never worked?

* skip that test
 2025-10-01 13:09:15 +03:00
b1tg
ac3d457d5e rangeify: TestReduceOpsConstFolding (#12397) 
Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2025-10-01 17:58:19 +08:00