Commit graph

5,763 commits

Author SHA1 Message Date
George Hotz
009b484dc0 bump 2026-06-17 16:22:13 -07:00
George Hotz
0b5796e3c8 fix gemm group + END shape 2026-06-17 16:16:29 -07:00
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
chenyu
1e08c0a07c
remove NOOP from AFTER with multiple srcs (#16646) 2026-06-17 14:35:02 -04:00
chenyu
1acc40600d
indexing an after with all fully invalid stores is invalid (#16643)
* indexing an after with all fully invalid stores is invalid

* typing cast
2026-06-17 11:06:36 -04:00
George Hotz
d631716858
remove const without STACK (#16639)
* remove const without STACK

* fix GEP rewrite

* fix null tests

* fix openpilot regression

* it's 10 in CI
2026-06-16 21:25:42 -07:00
wozeparrot
36f6d1b064
gemm: fix bf16 atb for mp sharding (#16637) 2026-06-16 15:58:47 -07:00
qazal
1cb6b88d37
viz: show contents of vconst (#16636)
* failing test

* render vconst

* simpler test

* reorder
2026-06-17 02:31:03 +09:00
chenyu
f0998e9bba
Revert "invalid clone is anonymous buffer" (#16613) (#16633) 2026-06-16 08:27:48 -04:00
qazal
7d2b0b697d
simple failing test for invalid extra E kernel (#16632)
* simple failing test for invalid extra E kernel

* 6 kernels
2026-06-16 17:57:44 +09:00
chenyu
efd03d7153
invalid clone is anonymous buffer [PR] (#16613) 2026-06-15 20:14:26 -04:00
George Hotz
41aa2fe119
test_gemm needs .clone() on eye (#16629) 2026-06-15 12:48:27 -07:00
George Hotz
b1fb39502d delete that test 2026-06-14 09:42:58 -07:00
chenyu
5d5ead78da
inline unique_const in invalids [PR] (#16612) 2026-06-13 10:14:32 -04:00
Sieds Lykles
b00dd754a9
Remove if-condition from nested div rule [pr] (#16611)
* add rules and test

* trigger [pr]
2026-06-13 15:47:21 +02:00
nimlgen
c43091a464
fix missing cast in cstyle (#16608)
* fix missing cast in cstyle

* x

* x
2026-06-13 10:04:06 +03:00
Christopher Milan
8862c7549c
new-style dcache_flush (#16602) 2026-06-12 22:25:08 -04:00
chenyu
aa32d309db
fix rangeify indexing for pad/reduce (#16599) 2026-06-12 20:26:15 -04:00
qazal
b2e95b2db3
rangeify: no copies for write+read of same slice (#16585)
* failing test

* cleaner failing tests

* assign and read of same slice shouldn't create copies

* err in the changes

* shrink with no overlapping regions in dest is fine
2026-06-13 02:19:47 +09:00
Philip Sinitsin
76c10cd635
jit: don't memplan buffers reachable from live tensors (#16588)
The memory planner was suballocating BUFFERs created during JIT capture that are still referenced by external lazy tensor graphs, like the .grad tensors assigned by backward(). The replay then only writes the arena slices, so realizing such a tensor after the call reads freshly allocated memory and silently returns zeros. Hold every BUFFER reachable from a live Tensor instead of only the parameters of the return value; true internals are still planned. Fixes #16571.
2026-06-12 17:51:54 +03:00
qazal
4d34590b7d
llama: less E kernels (#16517) 2026-06-12 19:49:25 +09:00
qazal
12f4cf0e49
rename amd/test_custom_kernel.py to test_asm_kernel (#16586)
* rename amd/test_custom_kernel.py to test_asm_kernel

* update
2026-06-12 16:11:01 +09:00
George Hotz
b8aec4cce7
port x86 to new_style (fable slop) and now everything is new style (#16581)
* port x86 to new_style (fable slop)

* don't change ops

* port NIR to new_style (fable)

* lil cleanup

* fix tests, and remove new_style
2026-06-11 21:09:34 -07:00
chenyu
762f50bd52
move gradient.py to mixin/ [PR] (#16583) 2026-06-11 23:58:21 -04:00
chenyu
a2cec397f3
UOp cast and bitcast takes DTypeLike [PR] (#16582)
* UOp cast and bitcast takes DTypeLike [PR]

match Tensor

* fix type
2026-06-11 22:38:54 -04:00
Christopher Milan
4d893f626a
move a bunch of test_schedule to null (#16578) 2026-06-11 20:26:34 -04:00
chenyu
5f1e2d3900
PADTO pads Invalids (#16562) 2026-06-11 16:54:26 -04:00
qazal
a83710396c
support mselect input to CALL, less kernels in allreduce (#16567)
* support mselect input to CALL, less kernels in allreduce

* resolve mstack
2026-06-11 18:10:47 +09:00
qazal
21f1101691
add allreduce kernel count test (#16566) 2026-06-11 15:54:12 +09:00
wozeparrot
c38d6a7e3a
mxfp8 part 2 (#16561) 2026-06-10 23:36:11 -07:00
George Hotz
7e6d617935
addrspace cleanups (#16565)
* addrspace cleanups

* bumps

* eh, relax a little
2026-06-10 15:57:18 -07:00
qazal
34481830f1
rangeify: fix cost function for AFTER(out, CALL) (#16559)
* simple failing test

* fix rangeify cost function

* new ops count
2026-06-10 17:30:50 +09:00
Christopher Milan
f9d88d3c3a
fix race in test_quantize_onnx (#16555) 2026-06-09 18:39:48 -04:00
wozeparrot
2bdc360606
gemm: mxfp8 hipkittens gemm (#16541)
* gemm: mxfp8 hipkittens gemm

* feat: update hipkittens

* feat: kernel signature

* clean: just kernel

* feat: from tinygrad

* feat: test

* fix: add back utils

* clean: no diff

* clean: no diff
2026-06-09 15:20:05 -07:00
chenyu
3f053a3370
move functional part of rand to RandMixin (#16551) 2026-06-09 09:40:48 -04:00
qazal
598cc13ad2
more readable null graph profile in VIZ (#16548)
* more readable null graph profile in VIZ

* change

* fix flaky test
2026-06-09 18:35:05 +09:00
qazal
d18ad49f20
fix flaky test_disktensor (#16549) 2026-06-09 18:23:22 +09:00
qazal
fa400f9790
less E kernels in all2all (#16546) 2026-06-09 13:51:57 +09:00
qazal
b8931440ae
add all2all schedule test (#16545) 2026-06-09 12:41:35 +09:00
Christopher Milan
4e2e2e9956
ocelot: use c.DLL (#16540) 2026-06-08 21:27:28 -04:00
chenyu
11fee53527
RandMixin [PR] (#16543) 2026-06-08 19:11:28 -04:00
chenyu
12764161c9
UOp.shard support axis=None [PR] (#16538)
match Tensor
2026-06-08 11:36:50 -04:00
chenyu
ebc5390c9a
advance indexing to mixin [PR] (#16532) 2026-06-08 09:24:49 -04:00
George Hotz
0336ba8eb1
buffer param arg + dsp fixups (#16528) 2026-06-07 12:07:00 -07:00
chenyu
90b556ca48
move gradient to mixin [PR] (#16526) 2026-06-07 00:05:02 -04:00
chenyu
4e7c6260b0
clean up test_tesnor_uop_mixin (#16525)
most of those don't have UNIQUE anymore
2026-06-06 23:25:44 -04:00
George Hotz
2a2f81dd3d
remove ANON from addrspace, refactor marg (#16523)
* remove ANON from addrspace, refactor marg

* as_shape

* as_shape is cached
2026-06-06 09:49:09 -07:00
Christopher Milan
8c0ba1da5c
cleanup more from test/backend (#16521) 2026-06-05 18:38:46 -04:00
Christopher Milan
9b0f75622c
many jit tests belong in unit (#16508) 2026-06-04 21:36:53 -04:00
chenyu
bb407d8b3c
fix transform_precompiled_call for MULTI (#16510)
based on my understanding for https://github.com/tinygrad/tinygrad/pull/16084
2026-06-04 20:09:58 -04:00