DeepGEMM/deep_gemm
Chenggang Zhao 3a5539b7db Use c++20
2025-04-03 15:47:59 +08:00
..
include/deep_gemm Solve STSM bank conflict via padding and 3D TMA 2025-04-03 15:39:35 +08:00
jit Use c++20 2025-04-03 15:47:59 +08:00
jit_kernels Solve STSM bank conflict via padding and 3D TMA 2025-04-03 15:39:35 +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