George Hotz 8cbef912d2 move reshape to MathTraits (#13054) ...

* move reshape to MathTraits

* confirm it works in amd_uop_matmul

2025-11-02 12:56:15 +08:00

..

amd_seb

kernel4 and 5 in uops (#11411)

2025-07-28 19:35:48 -07:00

max_kernels

extra/gemm/max_matmul: start of custom kernels for GEMM (#6926)

2025-03-19 15:04:57 +08:00

.gitignore

fast amd gemm (#9318)

2025-03-03 12:01:14 +08:00

amd_matmul.py

kernel4 and 5 in uops (#11411)

2025-07-28 19:35:48 -07:00

amd_uop_matmul.py

move reshape to MathTraits (#13054)

2025-11-02 12:56:15 +08:00

amx.py

rename allocator methods to not conflict [pr] (#7788)

2024-11-20 00:10:29 +08:00

cuda_matmul.py

rename allocator methods to not conflict [pr] (#7788)

2024-11-20 00:10:29 +08:00

fuzz_matmul.py

acc_dtype -> dtype (#9402)

2025-03-10 16:05:30 -04:00

gemm.c

only 62 gflops (#2629)

2023-12-05 13:28:24 -08:00

gemm.py

only 62 gflops (#2629)

2023-12-05 13:28:24 -08:00

halide_gemm.py

add halide example (#10980)

2025-06-26 16:14:57 -07:00

hip_matmul.py

rename allocator methods to not conflict [pr] (#7788)

2024-11-20 00:10:29 +08:00

intel_xmx.py

ops_gpu -> ops_cl (#12103)

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

jax_pmatmul.py

jax parallel matmul example

2023-11-28 13:48:11 -08:00

max_matmul.py

auto-select available compilers (#12094)

2025-09-10 19:52:01 +03:00

metal_conv.py

create engine folder and move code (#3948)

2024-03-26 20:38:03 -07:00

metal_matmul.py

compile fixes (#10442)

2025-06-06 18:38:37 -04:00

metal_matvec.py

compile fixes (#10442)

2025-06-06 18:38:37 -04:00

metal_uop_matmul.py

matmul example on metal showing off tensor core (#13033)

2025-10-31 19:40:36 +08:00

mi350x_uop_matmul.py

add CONTRACT support to UOp programs (#13043)

2025-11-01 19:11:32 +08:00

mlx_matmul.py

mlx benchmark, a lil slower than tg

2023-12-05 19:00:43 -08:00

real_pmatmul.py

pmatmul example + GB/s bugfix [run_process_replay] (#5974)

2024-08-07 22:32:11 -07:00

simple_conv.py

acc_dtype -> dtype (#9402)

2025-03-10 16:05:30 -04:00

simple_matmul.py

cuda fp8 (#12782)

2025-10-21 15:05:25 -04:00

simple_matvec.py

acc_dtype -> dtype (#9402)

2025-03-10 16:05:30 -04:00

tf_gemm.py

Add tensorflow GEMM benchmark script (#1000)

2023-06-18 10:57:45 -07:00

tinygrad_nv_matmul.py

update Kernel API in tests + move optimize_local_size (#11907)

2025-08-28 15:12:47 -07:00

torch_gemm.py

fp16 acc

2025-11-02 12:53:04 +08:00

triton_nv_matmul.py

auto-select available compilers (#12094)

2025-09-10 19:52:01 +03:00

tvm_gemm.py

move opt under codegen (#11569)

2025-08-07 14:19:17 -07:00