George Hotz a73d8717f3 fast amd gemm (#9318) ...

* 50 TFLOP AMD gemm

* add lds tiling

* register tiling

* flip locals

* work

* comment

* remove those

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

..

.gitignore

fast amd gemm (#9318)

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

amd_matmul.py

fast amd gemm (#9318)

2025-03-03 12:01:14 +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

wmma: widen TC usage in search by using PADTO on TC axes when possible (#4216)

2024-04-22 16:50:31 -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

hip_matmul.py

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

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

intel_xmx.py

Intel XMX Tensor Core Support (#5622)

2024-08-16 09:19:21 -07:00

jax_pmatmul.py

jax parallel matmul example

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

kernel8_batched_gmem.s

fast amd gemm (#9318)

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

metal_conv.py

create engine folder and move code (#3948)

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

metal_matmul.py

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

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

metal_matvec.py

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

2024-11-20 00:10:29 +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

wmma: refactor to remove wmma_func and create TC funcs as needed (#3945)

2024-03-27 16:43:09 -04:00

simple_matmul.py

fix for int plus minor cleanup (#8650)

2025-01-17 22:30:39 -05:00

simple_matvec.py

extra/gemm/simple_matvec: add simple_matvec.py (#4021)

2024-03-31 16:38:52 -04:00

tf_gemm.py

Add tensorflow GEMM benchmark script (#1000)

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

tinygrad_nv_matmul.py

remove graph [pr] (#7085)

2024-10-16 11:40:07 +08:00

torch_gemm.py

speed docs + upgrades [pr] (#8964)

2025-02-08 17:28:52 +08:00

triton_nv_matmul.py

BufferSpec and ProgramSpec [pr] (#7814)

2024-11-21 12:18:05 +08:00

tvm_gemm.py

CLANG -> CPU (#9189)

2025-02-20 18:03:09 -05:00