Francis Lam 4273aabe31 extra/gemm: add a simple_conv.py along with correctness check (#3236) ...

* extra/gemm: add a simple_conv.py along with correctness check

The goal is to easily test tensor core triggering situations

* test: add tests for acc_dtype handling and fixed typing

2024-01-26 19:06:57 -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 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 device to device.py (#2466)

2023-11-27 11:34:37 -08:00

metal_matmul.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

metal_matvec.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08: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

fix tvm gemm example

2023-10-08 05:57:41 -07:00