Zhean Xu 4499c4ccbb Refactor MMA template with CUTLASS (#87) ... * Refactor MMA with cutlass * Update README.md --------- Co-authored-by: Zhean Xu <xza@deepseek.com>		2025-04-14 17:06:49 +08:00
..
include/deep_gemm	Refactor MMA template with CUTLASS (#87)	2025-04-14 17:06:49 +08:00
jit	Use swizzling instead of padding (#86)	2025-04-14 15:20:58 +08:00
jit_kernels	Use swizzling instead of padding (#86)	2025-04-14 15:20:58 +08:00
__init__.py	fix typo	2025-02-26 18:37:22 +08:00
utils.py	Correctly flush L2, as reconstructing the tensors on every iteration effectively put them in the L2, and gave the GPU enough idle time to avoid thermal throttling in a potentially unrealistic way.	2025-03-15 20:46:24 +00:00