DeepGEMM/deep_gemm
yukuai26 95e81b3dd6
Indivisible TMA (#90)
Fix indivisible shapes for TMA multicast

---------

Co-authored-by: yukuai <yukuai@deepseek.com>
Co-authored-by: Chenggang Zhao <chenggangz@deepseek.com>
2025-04-23 14:55:14 +08:00
..
include/deep_gemm Indivisible TMA (#90) 2025-04-23 14:55:14 +08:00
jit Use swizzling instead of padding (#86) 2025-04-14 15:20:58 +08:00
jit_kernels Indivisible TMA (#90) 2025-04-23 14:55:14 +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