mirror of https://github.com/commaai/tinygrad.git
bbb0ad4800
* 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 |
||
|---|---|---|
| .. | ||
| .gitignore | ||
| amx.py | ||
| cuda_matmul.py | ||
| fuzz_matmul.py | ||
| gemm.c | ||
| gemm.py | ||
| hip_matmul.py | ||
| intel_xmx.py | ||
| jax_pmatmul.py | ||
| metal_conv.py | ||
| metal_matmul.py | ||
| metal_matvec.py | ||
| mlx_matmul.py | ||
| simple_conv.py | ||
| simple_matmul.py | ||
| simple_matvec.py | ||
| tf_gemm.py | ||
| torch_gemm.py | ||
| tvm_gemm.py | ||