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

..

__init__.py

fix test extra issue (#3159)

2024-01-17 11:58:08 -08:00

test_asm_gemm.py

grad_b uses custom gemm (#14550)

2026-02-05 15:22:27 +09:00

test_bench_log.py

update tests, WEBGPU=1 pytest . passes (#14089)

2026-01-10 00:03:02 -05:00

test_cfg_viz.py

remove DEFINE_GLOBAL (it is PARAM now) (#14488)

2026-02-02 14:56:37 +08:00

test_export_model.py

Use matching JS TypedArray for buffer dtype (#8080)

2024-12-06 14:52:23 +01:00

test_f16_decompress.py

only run test_u32_to_f16 if it supports fp16 (#10277)

2025-05-13 11:16:14 -04:00

test_fp8_linear.py

train bert with fp8 (#13874)

2026-01-09 09:21:59 -05:00

test_hevc.py

test_hevc: do not download the whole file (#13531)

2025-12-02 21:31:28 +03:00

test_lr_scheduler.py

fix some slow tests [pr] (#8979)

2025-02-09 15:57:04 +08:00

test_mockgpu.py

move mockgpu to tests [pr] (#8396)

2024-12-24 23:48:02 +03:00

test_tk.py

fa: faster (#14453)

2026-02-02 21:34:17 -08:00