Francis Lam 7c5729a3bd wmma: refactor to remove wmma_func and create TC funcs as needed (#3945) ...

* wmma: refactor to remove wmma_func and create TC funcs as needed

* test_linearizer: disable bf16 CUDA during emulation testing

* cstyle: clean up creation of CUDA vec dtypes

* extra/gemm: add option to accumulate to bfloat16

* cleanups

* benchmark: add CUDA bfloat16 matmul

* more cleanups

2024-03-27 16:43:09 -04:00

13 KiB

Raw Blame History

View Raw