github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-21 21:08:07 -05:00
Code Issues Packages Projects Releases Wiki Activity
10,633 Commits 204 Branches 14 Tags
multi_range
Commit Graph

10633 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
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
George Hotz
60e52fbe36 support opts in contig, simpler (#12400) 2025-10-01 17:20:04 +08:00
chenyu
6c95b1f39d explicitly set device for CI unit test (#12399) 2025-10-01 05:16:54 -04:00
chenyu
6ba8bf282f skip test_masked_select for RANGEIFY PYTHON (#12395) 2025-10-01 04:13:31 -04:00
chenyu
689ab9151b more RANGEIFY tests (#12393) 
would have caught the load alt regression without adding too many tests
 2025-10-01 03:43:58 -04:00
chenyu
adc8c3b28f Revert "load alt value with cast (#12384)" (#12392) 
This reverts commit 05e91a248d.
 2025-10-01 03:20:04 -04:00
b1tg
154d114364 rangeify: fix abstractions2.py (#12386) 
* rangeify: fix abstractions2.py

* tests

* lint

* only abstractions2

* base
 2025-10-01 09:58:56 +03:00
George Hotz
fe96c8d345 add HALF flag to tinygrad.apps.llm 2025-10-01 14:44:59 +08:00
George Hotz
f205352cd7 remove ranges with 1s (#12388) 
* use op_in_parents

* remove the ranges of 1

* fix CL image thing

* fix realize
 2025-10-01 14:43:29 +08:00
qazal
90b1c0dd96 rangeify: test_where_fold kernel count (#12379) 
* rangeify: test_where_fold kernel count

* get these from the index

* replace ranges

* fine

* movement ops

* diff

* better
 2025-10-01 09:35:12 +03:00
b1tg
42748ccb92 rangeify: fix test_prequant_conv2d_1x1 (#12391) 2025-10-01 02:33:47 -04:00
Sieds Lykles
05e91a248d load alt value with cast (#12384) 
* add or_casted

* add tests and fix old tests

* cast load

* move that to pm_render
 2025-10-01 07:14:26 +02:00
qazal
714500edfd viz: add font-weight to OffscreenCanvas config (#12390) 2025-10-01 08:08:47 +03:00
b1tg
57ad46c6e4 rangeify: increase atol for test_two_binops_no_rerun passing on real windows machine (#12389) 
CPU_LLVM=1
 2025-10-01 00:56:45 -04:00
George Hotz
e02da8f5ac use op_in_parents (#12385) 2025-10-01 12:37:29 +08:00
chenyu
0662946fac atol in test_two_binops_no_rerun (#12387) 
for RANGEIFY LLVM
 2025-10-01 00:05:47 -04:00
b1tg
da52006bde rangeify: fix test_scatter_reduce (#12380) 
* rangeify: fix test_scatter_reduce

* ext_vector_type

* set alignment=1 on boolean
 2025-09-30 23:26:36 -04:00
George Hotz
1c1b4d14e9 minor cleaups in rangeify (#12382) 
* minor cleaups in rangeify

* op_in_parents

* don't use toposort

* Revert "don't use toposort"

This reverts commit 257d8e2529.
 2025-10-01 11:19:48 +08:00
wozeparrot
4204edc60b feat: skip test_long (#12383) 2025-09-30 20:07:39 -07:00
chenyu
8def8145e4 ALLOWED_KERNEL_COUNT openpilot 0.9.4 with RANGEIFY (#12381) 2025-09-30 22:58:59 -04:00
George Hotz
4c9a930de2 rangeify attn tests (#12377) 2025-10-01 09:59:19 +08:00
qazal
26247573e1 rangeify multi tests on gpu (#12376) 
* rangeify multi tests on gpu

* fix limit_bufs
 2025-10-01 04:53:04 +03:00
qazal
f2eb92948d rangeify: ban view pushing (#12371) 
* rangeify: ban view pushing

* don't shape INDEX

* fix the codegen cache

* make space
 2025-10-01 04:37:52 +03:00
George Hotz
a128fa0f8a removing double reshapes was wrong (#12375) 2025-10-01 09:25:35 +08:00
hooved
969a1b35ca LR scheduler for Stable Diffusion mlperf training (#12201) 
* add lr scheduler for stable diffusion training

* add lr scheduler test

* rerun ci

* rerun CI

* use np for testing

* move test to CI path

* remove unneeded copy
 2025-09-30 21:21:08 -04:00
George Hotz
9ef319f349 bad conv in rangeify (#12373) 
* bad conv with broken rangeify

* no maxpool needed

* add empty_like

* typo

* no self

* issue remains for test
 2025-10-01 08:56:22 +08:00
nimlgen
080b26e7d7 use suppress_finalizing to not mute all exceptions (#12372) 2025-09-30 21:24:31 +03:00
George Hotz
44558a37f7 fix some rangeify tests (#12370) 
* fix bad range merges

* fix rng

* fix uop gc

* fix some rangeify tests

* now that needs rangeify 2 also
 2025-09-30 20:12:08 +08:00
nimlgen
2c397eb2a2 rangeify: buf limit (#12336) 
* limit bufs

* g

* fix buffer limit

* um?

* fix

* only these?

* typo

* f

* cleaner
 2025-09-30 14:59:47 +03:00
George Hotz
a83f219253 fix bad range merges (#12368) 
* fix bad range merges

* fix rng

* fix uop gc
 2025-09-30 19:30:21 +08:00
qazal
a95159d579 remove TestShapeSpec, it relies on ShapeTracker [pr] (#12369) 2025-09-30 14:20:35 +03:00
George Hotz
9cf5e66899 minimal rangeify stable diffusion fix (#12367) 
* minimal rangeify stable diffusion fix

* more minimal
 2025-09-30 18:48:35 +08:00
chenyu
b4a4817c9c fix rangeigy test_linalg (#12365) 2025-09-30 06:28:35 -04:00
qazal
de1d562b69 rangeify: update test_pickle asserts (#12366) 
* realized exists on the base

* use is_realized
 2025-09-30 13:27:41 +03:00
b1tg
c9ef5d8fe5 rangeify: fix test_tensor_index_overflow (CPU_LLVM=1) (#12362) 
* rangeify: fix test_tensor_index_overflow (CPU_LLVM=1)

* add test

---------

Co-authored-by: b1tg <b1tg@users.noreply.github.com>
 2025-09-30 05:55:15 -04:00