Caleb Bunch 0b1fc5888a fix 'Import Error: cannot import name compile_cuda from tinygrad.runtime.ops_cuda' error in extra/gemm/cuda_matmul.py (#3531)		2024-02-28 17:15:32 -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	fix 'Import Error: cannot import name compile_cuda from tinygrad.runtime.ops_cuda' error in extra/gemm/cuda_matmul.py (#3531)	2024-02-28 17:15:32 -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