chenyu
3e2b7803e6
view assign replaces at buffer identity ( #15298 )
...
matches what functions capture
2026-03-16 19:58:38 -04:00
George Hotz
476276f4b4
support grads on tuples ( #15287 )
...
* support grads on tuples
* simpler
* grad_fxn works
* cleanups
* unused
2026-03-16 17:39:34 +08:00
George Hotz
08662bc4ab
add TUPLE/GETTUPLE, simple tests pass ( #15286 )
...
* simple tuple stuff passes
* resolved
2026-03-16 15:06:02 +08:00
qazal
4445f50356
viz: variable duration rdna barriers ( #15277 )
...
* viz: variable length rdna barriers
* work
* tiny changes
* simple wave simd test
* small wave sync test
* good multi barrier bug find
* simple fix
* wave_sync asserts
* rdna4 work
* more rdna4
* find more bugs in my model
* it's so much simpler
* wave_sync tests duration
* r4
* should just call this rdna4
2026-03-16 06:06:19 +09:00
qazal
5cd1daa3bc
cdna asm_gemm in one file, remove old rdna3 asm ( #15281 )
2026-03-16 04:32:30 +09:00
chenyu
cd14e8e64b
allocations contiguous is store+after ( #15280 )
2026-03-15 11:58:40 -04:00
qazal
7b6211fdd7
sqtt: remove discover_ops script ( #15279 )
2026-03-15 22:17:06 +09:00
qazal
3858bfc83d
sqtt: CDNA inst decodes ( #15274 )
...
* sqtt: CDNA inst decodes
* JUMP packets other way
* cdna insts
* r3
* r4
* lds from simd1 and simd2
2026-03-14 21:03:46 +09:00
Christopher Milan
9047249a7c
m.where(x.pad_to(m.shape), Invalid) ranges shrink ( #15275 )
2026-03-14 07:26:36 -04:00
Christopher Milan
dabdc986df
shrink guarded ranges, try 2 ( #15272 )
2026-03-14 04:24:05 -04:00
Christopher Milan
7cf4b16c91
Revert "shrink guarded ranges" ( #15271 )
2026-03-14 03:44:38 -04:00
Christopher Milan
d9951e2f8e
shrink guarded ranges ( #15263 )
2026-03-14 03:38:48 -04:00
qazal
4d60312f7f
viz: asm python dsl syntax highlighting ( #15259 )
2026-03-14 06:37:43 +09:00
qazal
6209ddfc90
viz: improve disasm of s_code_end ( #15258 )
...
* viz: improve amd disasm of s_code_end
* better tests
* order was good
2026-03-14 03:31:14 +09:00
Sieds Lykles
4b59083d7c
assign into empty works ( #15256 )
2026-03-13 10:24:29 -04:00
qazal
60b1b908c6
sqtt: CDNA layout header packet is the same size ( #15255 )
2026-03-13 22:28:24 +09:00
chenyu
018c01508d
test case for call precompile multi ( #15254 )
2026-03-13 06:28:43 -04:00
qazal
d893b14193
sqtt: update cdna packet names ( #15243 )
...
* sqtt: update cdna packet names
* change
* order
2026-03-13 08:49:09 +09:00
chenyu
90b7f4341d
failed two level divmod recombine case ( #15233 )
2026-03-12 04:04:36 -04:00
chenyu
842c978df3
remove staticmethod dtypes.max/min ( #15227 )
...
always use x.dtype.max/min
2026-03-11 23:11:24 -04:00
b1tg
18dc77ccab
add fp8 fnuz dtypes with PYTHON backend support ( #14945 )
...
* add fp8 fnuz dtypes with PYTHON backend support
* rm emu related change
* clarify fp8 fnuz zero handling
* Revert "rm emu related change"
This reverts commit efa4763c22 .
---------
Co-authored-by: b1tg <b1tg@users.noreply.github.com >
Co-authored-by: chenyu <chenyu@fastmail.com >
2026-03-11 22:30:18 -04:00
George Hotz
4f3f55328b
do not patch on invalid tensor tests ( #15226 )
...
* do not patch on invalid tensor tests
* cleanup
2026-03-12 09:35:20 +08:00
qazal
d3eef70162
viz: render shader clock frequency graph ( #15197 )
2026-03-12 01:32:49 +09:00
Christopher Milan
2fb8a7f60f
fix test_invalid_tensor when before values are nan ( #15215 )
2026-03-10 23:51:19 -04:00
chenyu
fce87f19a8
better fold_add_divmod_recombine ( #15214 )
2026-03-10 23:24:22 -04:00
chenyu
df8deec949
test for nest_by_factor selection ( #15213 )
2026-03-10 22:41:31 -04:00
chenyu
be6b0bce1f
variations of (x%c)+(x//c)*c ( #15212 )
...
put those into one function
2026-03-10 22:41:14 -04:00
qazal
a408d90f4f
viz: always detect sqtt packet overlaps, add timeline tests ( #15211 )
...
* test
* work
* it's called CALL, better assert
* qol
* row_ends
2026-03-11 05:32:38 +09:00
Christopher Milan
25d86ec9e1
start using Invalid in image_conv2d ( #15208 )
2026-03-10 07:11:06 -04:00
chenyu
8389a8d7c5
remove_nested_mod can work with negative ( #15205 )
2026-03-10 03:10:08 -04:00
Christopher Milan
ffaafd391a
Invalid in Tensor ( #15154 )
2026-03-10 02:49:54 -04:00
chenyu
68c7c3ca84
divmod test_gcd_with_remainder ( #15204 )
...
test cases for gcd_with_remainder
2026-03-09 23:51:47 -04:00
chenyu
a53187eef7
fix TestPartialAssignToSharedBuffer ( #15202 )
...
bufferize_to_store issue with assign
2026-03-09 23:14:23 -04:00
chenyu
60215deb60
tiebreak in fold_divmod_congruence ( #15190 )
...
need to try both direction
2026-03-09 03:40:39 -04:00
chenyu
a8d8351e5a
match IDIV and MOD in nest_by_factor ( #15188 )
2026-03-09 00:50:38 -04:00
chenyu
83b80da8f3
even more divmod recombine ( #15163 )
2026-03-08 23:52:26 -04:00
qazal
25e82a9aca
viz: exclude redundant traceback from SDMA ( #15185 )
...
* viz: exclude redundant traceback from SDMA
* ctx
* cpu_profile
2026-03-09 05:12:14 +09:00
nimlgen
6ac99fd4c9
memplanner opt copy bufs ( #15110 )
...
* mtp
* x
* tests
* ss
* simp
* less slop
* x
* cleaner
* rm
* m
* c
* x
* f
2026-03-08 22:28:01 +03:00
nimlgen
633264feae
am: flush sdma pipeline ( #15184 )
...
* am: flush sdma pipeline
* f
* f
* fix
2026-03-08 20:27:56 +03:00
b1tg
891a73befc
llm: fix chunked prefill ( #15182 )
...
* llm: fix chunked prefill
* less lines
---------
Co-authored-by: b1tg <b1tg@users.noreply.github.com >
2026-03-07 22:08:31 +08:00
Ananta Ranganathan
5bdad8ee41
update mxfp4 tests to use the same patterns as the others ( #15177 )
...
* update mxfp4 tests to use the same patterns as the others
* fix typo in test call not sure how it committed
2026-03-06 13:21:40 -05:00
Ananta Ranganathan
5c50035e0d
avoid using arithmetic for mxfp4 ( #15172 )
...
* avoid using arithmetic for mxfp4
* update tests to use assert equal
* no longer todo
2026-03-06 11:17:56 -05:00
Roelof van Dijk
4ed8bb7445
tie break for divmod ( #15169 )
2026-03-06 08:05:38 -05:00
qazal
83f1faa142
sqtt: update CDNA wave packet field, start unskipping tests ( #15168 )
...
* correct field names
* packet types
* packet 5 is regc
* test skips
2026-03-06 21:37:44 +09:00
Christopher Milan
7810be8d3c
compile QCOM without opening device ( #15165 )
...
Co-authored-by: Comma Device <device@comma.ai >
2026-03-06 06:24:27 -05:00
George Hotz
6fd18ef875
rename CAT to VCAT ( #15167 )
2026-03-06 18:46:28 +08:00
Roelof van Dijk
059c6326c0
metal uint32 icb offset overflow ( #15156 )
...
* metal uint32 icb offset overflow
fix: diff
supports_exec_item
GraphRunner.supports_exec_item
tests
fix: can't import on non-metal
stricter
* also test the non-metal buffer case
* imports on non-mac
2026-03-06 00:54:39 +03:00
chenyu
da61088ca4
more divmod recombine ( #15162 )
2026-03-05 12:53:22 -05:00
chenyu
167a1d56a6
improve divmod folding ( #15148 )
...
canonicalize to div than mod which enables more simplifcation
2026-03-05 10:07:36 -05:00
qazal
5bf542469d
viz: python traceback for USER device ( #15160 )
...
* start
* ux
* unittests
2026-03-05 20:22:09 +09:00