7 commits

Author	SHA1	Message	Date
qazal	80b0119cef	llama: add new asm gemm shape (#14611) ... * llama: add new asm gemm shape * work * cleanup * half dtype * more comment	2026-02-10 00:34:29 +09:00
qazal	087dab4c3b	gemm/asm: split out cdna tests from CI (#14619) ... * gemm/asm: split out cdna tests from CI * reorder * work	2026-02-08 21:33:42 +09:00
George Hotz	183d38b128	remove CUSTOM_KERNEL / directly construct it (#14604) ... * remove CUSTOM_KERNEL / directly construct it * clean that up * simpler multi * custom kernel spec * remove Kernel * fix multi * use sharded shape * explicit regression test	2026-02-08 18:43:33 +08:00
qazal	cf73d7e2a7	hotfix: disable slower asm gemm shape from llama seqlen 8192 (#14582)	2026-02-06 15:05:19 +09:00
George Hotz	6cbcf98627	KernelInfo is required on get_program (#14571) ... * rangeify always adds KernelInfo * fix tests * skip flaky test	2026-02-06 10:49:27 +08:00
George Hotz	43e7eda4e7	grad_b uses custom gemm (#14550) ... * grad_b uses custom gemm * fix multi backward, acc is in float32 * test_gemm_batched * square gemm --------- Co-authored-by: qazal <77887910+Qazalin@users.noreply.github.com> Co-authored-by: qazal <qazal.software@gmail.com>	2026-02-05 15:22:27 +09:00
qazal	f9cfb64cd9	test asm_gemm in CI (#14551) ... * test asm_gemm in CI * default float16 * use a smaller shape for multi * smaller size * smaller for CI * smaller for ci * need half	2026-02-05 13:32:22 +09:00

Renamed from extra/gemm/asm/cdna/test_asm_gemm.py (Browse further)