DeepGEMM/deep_gemm
A-transformer b0b9e03345
refactor the loop if/else check
Keeps DivisibleK and NotDivisibleK logic explicit
Ensures clear separation between the bulk iterations and the last special case
Easier to modify later if conditions change
2025-02-27 22:23:53 +04:00
..
include/deep_gemm refactor the loop if/else check 2025-02-27 22:23:53 +04:00
jit Fix TMA store bugs and code format 2025-02-27 17:57:21 +08:00
jit_kernels Update get_best_configs 2025-02-27 23:18:52 +08:00
__init__.py fix typo 2025-02-26 18:37:22 +08:00
utils.py Initial commit 2025-02-25 22:52:41 +08:00