tinygrad/extra/gemm
Yixiang Gao fde44aed76
update hip_matmul with new abstraction (#2605)
2023-12-04 13:37:10 -08:00
..
fastvits move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
.gitignore updates from the chonker branch 2022-11-07 21:12:08 -08:00
amx.py fixes (#1893) 2023-09-22 07:20:27 +08:00
cuda_matmul.py Update cuda_matmul.py (#2495) 2023-11-28 19:46:01 -08:00
gemm.c Revert "update editorconfig, enforce via CI (#1343)" (#1380) 2023-07-31 10:35:50 -07:00
gemm.py fixes (#1893) 2023-09-22 07:20:27 +08:00
gemv_845.py openpilot compile2 (#1977) 2023-10-15 20:39:46 -07:00
hip_matmul.py update hip_matmul with new abstraction (#2605) 2023-12-04 13:37:10 -08:00
jax_pmatmul.py jax parallel matmul example 2023-11-28 13:48:11 -08:00
metal_conv.py move device to device.py (#2466) 2023-11-27 11:34:37 -08:00
metal_matmul.py update metal matmul and matvec for compile api (#2238) 2023-11-08 08:08:35 -08:00
metal_matvec.py move device to device.py (#2466) 2023-11-27 11:34:37 -08:00
simple_matmul.py wmma: clean up to make WMMA arg order consistent (#2014) 2023-10-07 17:45:40 -07:00
tf_gemm.py Add tensorflow GEMM benchmark script (#1000) 2023-06-18 10:57:45 -07:00
torch_gemm.py faster RDNA assembly backend (#990) 2023-06-16 12:06:38 -07:00
tvm_gemm.py fix tvm gemm example 2023-10-08 05:57:41 -07:00