13 Commits

Author	SHA1	Message	Date
qazal	9f2a578e26	unskip TestCall.test_call_gemm_uop [pr] (#15786)	2026-04-17 16:18:51 +03:00
George Hotz	d24466c844	CALL with return value is FUNCTION (#15758) ... * CALL with return value is FUNCTION (GPT try) * cleanups	2026-04-16 13:25:07 +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	018c01508d	test case for call precompile multi (#15254)	2026-03-13 06:28:43 -04:00
George Hotz	fb43b415f9	fix symbolic shape call + chunked prefill (#15149) ... * fix precompile for symbolic shape * chunked prefill * cleaner * test that	2026-03-05 14:02:26 +08:00
George Hotz	ac1847cbf7	fully symbolic llm (#15097) ... * work * llm symbolic (almost) * work * revert that * llm sym * works * cleanups * cache tokens with the kv cache * cleanups * cleanups	2026-03-05 10:22:11 +08:00
George Hotz	01ddb4c267	add precompile to call (#15099) ... * add precompile to call * put get back * something * after structure * alt * keep it call * resolve call * resolve linear call * precompile works with llm * revert rangeify * color for debugging * getenv PRECOMPILE * clean up deco pattern * fully recursive sink scheduling * revert llama * fix SPEC=2	2026-03-03 22:32:42 +08:00
chenyu	06ef8a26b7	add a test case that triggers CALL passthrough_multi (#14887)	2026-02-19 10:45:40 -05:00
George Hotz	888819ee09	call autodiff gradient (#14510)	2026-02-03 13:51:02 +08:00
chenyu	02afae04f4	atol in test_call_gemm (#14480) ... flaky	2026-02-01 11:24:58 -05:00
George Hotz	838cd078bc	use atomics for embedding backward (#14400) ... * embedding is slow * failing * float is fine * null * it fails * simplify embedding with broadcasting * ATOMIC_ADD incoming * min change * simpler test * better test * fix test * real test * simpler * cleanups * types and names * _zero_kernel * grad multi * hack * none * multi unshard * more for call * don't tag in call * good * call_multi * call_multi wow claude is useless * embedding backward mutli test * test passes * fix as_param * shape_to_shape_arg * add clip * before cast * fix spec=2, use atomics	2026-01-30 18:10:59 +08:00
George Hotz	7a9dee4e50	add call/param UOps (#14433) ... * add call/param UOps * resolve call * skip that for now * grad on call * fix tests	2026-01-30 14:51:45 +08:00