Francis Lam bbb0ad4800 wmma: widen TC usage in search by using PADTO on TC axes when possible (#4216) ...

* wmma: widen TC usage in search by using PADTO on TC axes when possible

* test: start tests for the new padding TC behavior

* search: upgrade padded TC search to TC_OPT >= 2

* test: add behavior and correctness test for padded TC

added optional argument to apply_tensor_core to set TC_OPT level

* linearizer: add tests for the PADTO behvaior and docs

2024-04-22 16:50:31 -04:00

..

.gitignore

updates from the chonker branch

2022-11-07 21:12:08 -08:00

amx.py

update amx gemm (#3991)

2024-03-29 11:45:03 -04: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

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

extra/gemm/hip_matmul: fix to use new HSA devices and no headers (#3999)

2024-03-30 15:42:23 -04:00

intel_xmx.py

intel matmul (#3830)

2024-03-25 22:37:20 -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

create engine folder and move code (#3948)

2024-03-26 20:38:03 -07:00

metal_matvec.py

move GlobalCounter to helpers (#4002)

2024-03-30 00:30:30 -04:00

mlx_matmul.py

mlx benchmark, a lil slower than tg

2023-12-05 19:00:43 -08: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

test: add fuzz_matmul and better debugging for simple_matmul (#4199)

2024-04-16 23:40:31 -04: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

torch_gemm.py

faster RDNA assembly backend (#990)

2023-06-16 12:06:38 -07:00

tvm_gemm.py

LinearizerOptions -> CompilerOptions (#3978)

2024-03-28 17:50:23 -04:00