tinygrad/test/testextra
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