George Hotz a3c78d47b3 speed docs + upgrades [pr] (#8964) ... * add some docs about speed [pr] * better torch gemm * enable locals on llvm/clang * disable locals for beam speed on LLVM/CLANG * 0x20 alignment in llvm allows ymm use		2025-02-08 17:28:52 +08:00
..
.gitignore	updates from the chonker branch	2022-11-07 21:12:08 -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
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	create_schedule([x.lazydata]) -> x.schedule() in tests (#8449)	2024-12-31 03:15:52 +08:00