4 Commits

Author	SHA1	Message	Date
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