tinygrad/extra/gemm
George Hotz 2e60012bcf
move create schedule and delete old API (#3377)
* move create schedule and delete old API

* fix test multitensor
2024-02-12 18:10:45 +01: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 fix cuda matmul (#2725) 2023-12-12 07:59:31 -08: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 Fix hip_matmul gemm in extra (#3241) 2024-01-25 16:03:04 -08:00
jax_pmatmul.py jax parallel matmul example 2023-11-28 13:48:11 -08:00
metal_conv.py move graph.py and jit.py into features (#3376) 2024-02-12 17:34:34 +01:00
metal_matmul.py move graph.py and jit.py into features (#3376) 2024-02-12 17:34:34 +01:00
metal_matvec.py move graph.py and jit.py into features (#3376) 2024-02-12 17:34:34 +01:00
mlx_matmul.py mlx benchmark, a lil slower than tg 2023-12-05 19:00:43 -08:00
simple_conv.py extra/gemm: add a simple_conv.py along with correctness check (#3236) 2024-01-26 19:06:57 -08:00
simple_matmul.py wmma: enable METAL half tensor cores and clean up cstyle (#3095) 2024-01-12 16:25:28 -05: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 move create schedule and delete old API (#3377) 2024-02-12 18:10:45 +01:00