5,297 commits

Author	SHA1	Message	Date
George Hotz	34d2ef6447	Merge branch 'master' into precompile_backward	2026-03-17 15:17:41 +08:00
George Hotz	125d987886	no NOOPT	2026-03-17 15:14:25 +08:00
b1tg	856a839efc	llm: fix qwen3 moe topk renormalization (#15201)	2026-03-17 12:57:33 +08:00
George Hotz	9f591b42d1	add precompile backward support	2026-03-17 11:40:09 +08:00
George Hotz	3ff03be413	call always has tuple (#15297) ... * call always has tuple * fix pre-commit and simplify * update * fix * move that assert * tuple * fix multi * cleanups * fix merge	2026-03-17 10:58:46 +08:00
wozeparrot	674c760974	embedded bwd vocab shard (#15001) ... * fix: remove more multi from call * feat: embedding bwd vocab sharding * clean: unused import * clean: don't actually need this pattern	2026-03-16 19:37:16 -07:00
chenyu	02afb45f29	remove UOp.assign [pr] (#15300) ... * remove UOp.assign [pr] it's all store and after, UOp is immutable * fix test	2026-03-16 21:45:41 -04:00
qazal	33bd33e783	sqtt: add CDNA ops enum, show in viz (#15140)	2026-03-17 09:38:42 +09:00
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