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

..

device

Buffer.as_buffer -> Buffer.as_memoryview [pr] (#14535)

2026-02-04 11:31:11 -05:00

external

Buffer.as_buffer -> Buffer.as_memoryview [pr] (#14535)

2026-02-04 11:31:11 -05:00

mockgpu

assembly/amd: add RDNA4 support to emulator (#14341)

2026-02-02 21:35:59 +08:00

models

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00

null

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00

opt

Buffer.as_buffer -> Buffer.as_memoryview [pr] (#14535)

2026-02-04 11:31:11 -05:00

speed

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

testextra

grad_b uses custom gemm (#14550)

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

unit

merge as_buf into buf_uop [pr] (#14541)

2026-02-04 16:32:23 -05:00

web

remove stale / untested viz related files (#13785)

2025-12-21 16:42:48 +08:00

__init__.py

All devices are equal! (#196)

2020-12-15 23:44:08 -08:00

Dockerfile

new cloud is cloudy [pr] (#7631)

2024-11-11 20:18:04 +08:00

helpers.py

rename all DEFINE_GLOBAL to PARAM (#14511)

2026-02-03 15:09:38 +08:00

test_arange.py

use atomics for embedding backward (#14400)

2026-01-30 18:10:59 +08:00

test_const_folding.py

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00

test_custom_kernel.py

test schedule with multiple AFTER (#14449)

2026-01-30 15:59:00 -05:00

test_dtype_alu.py

decomp float16 to float32 (#14417)

2026-02-04 01:37:47 -05:00

test_dtype.py

decomp float16 to float32 (#14417)

2026-02-04 01:37:47 -05:00

test_edgecases.py

don't implicitly change dtype in assign (#14481)

2026-02-01 11:48:54 -05:00

test_fusion_op.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_gc.py

fix a few tests [pr] (#13498)

2025-11-29 13:43:45 -08:00

test_graph.py

Buffer.as_buffer -> Buffer.as_memoryview [pr] (#14535)

2026-02-04 11:31:11 -05:00

test_image_dtype.py

Buffer.as_buffer -> Buffer.as_memoryview [pr] (#14535)

2026-02-04 11:31:11 -05:00

test_interop.py

hotfix: interop example (#9237)

2025-02-25 10:32:00 +03:00

test_jit_cases.py

enumerate cases of Tensors in the JIT (#10548)

2025-05-28 11:51:27 -07:00

test_jit_footguns.py

raise if Tensor._buffer is called during jit (#14114)

2026-01-22 17:30:18 -05:00

test_jit.py

remove REAL_DEV [pr] (#14337)

2026-01-26 10:08:16 -05:00

test_kernel_cache.py

make method_cache account for compiler (#12156)

2025-09-13 17:00:11 +03:00

test_linearizer_dumb.py

rename all DEFINE_GLOBAL to PARAM (#14511)

2026-02-03 15:09:38 +08:00

test_linearizer.py

Buffer.as_buffer -> Buffer.as_memoryview [pr] (#14535)

2026-02-04 11:31:11 -05:00

test_multitensor.py

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00

test_nn.py

mark slow tests as slow instead of as CI (#13736)

2025-12-17 10:29:57 -04:00

test_ops.py

fix webgpu runtime types (#14402)

2026-01-28 10:37:25 -05:00

test_opt_gemm.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_optim.py

mark slow tests as slow instead of as CI (#13736)

2025-12-17 10:29:57 -04:00

test_outerworld_range.py

remove dtype from range, it will be dtypes.index soon [pr] (#11914)

2025-08-29 09:52:07 -07:00

test_outerworld.py

vmap on full model (#13340)

2025-11-18 16:06:06 -08:00

test_pickle.py

Buffer.as_buffer -> Buffer.as_memoryview [pr] (#14535)

2026-02-04 11:31:11 -05:00

test_profiler.py

fix test_dev_jitter_matrix (#14255)

2026-01-20 20:07:51 +03:00

test_quantize_onnx.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_randomness.py

assembly/amd: bug fixes for PYTHON_REMU (#14347)

2026-01-27 00:48:22 +08:00

test_rangeify.py

add device to local, fix PCONTIG=2 (#14266)

2026-01-21 22:12:18 +09:00

test_renderer_failures.py

Buffer.as_buffer -> Buffer.as_memoryview [pr] (#14535)

2026-02-04 11:31:11 -05:00

test_schedule.py

merge as_buf into buf_uop [pr] (#14541)

2026-02-04 16:32:23 -05:00

test_setitem.py

relax setitem target check (#14526)

2026-02-03 22:32:49 -05:00

test_softmax_fusion.py

remove ScheduleItem and merge it with ExecItem (#13759)

2025-12-19 17:04:24 -04:00

test_stunning.py

remove FUSE_ARANGE (#12511)

2025-10-08 04:54:07 -04:00

test_subbuffer.py

Buffer.as_buffer -> Buffer.as_memoryview [pr] (#14535)

2026-02-04 11:31:11 -05:00

test_symbolic_jit.py

don't allow jit input to be const (#14045)

2026-01-06 18:15:22 -05:00

test_symbolic_ops.py

minor type cleanups [pr] (#14408)

2026-01-28 14:11:50 -05:00

test_tensor_variable.py

fix test_symbolic_arange_sym_step (#13952)

2026-01-01 09:41:07 -05:00

test_tensor.py

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00

test_tiny.py

beautiful SQTT_PARSE=1 with color (#13428)

2025-11-23 01:05:14 -08:00

test_to_numpy.py

Apply ruff linting rules to tests (#2473)

2023-11-27 21:24:06 -08:00

test_transcendental.py

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

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

test_uops.py

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00

test_zero_copy.py

Buffer.as_buffer -> Buffer.as_memoryview [pr] (#14535)

2026-02-04 11:31:11 -05:00