3 Commits

Author	SHA1	Message	Date
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
Francis Lam	a26090d404	search: change to use "spawn" and limit the number of tasks per child (#3862) ... also clean up some examples to use __main__ and not initialize resources outside of main	2024-03-21 21:23:36 -07:00
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