DeepGEMM/deep_gemm/jit_kernels
2025-05-07 13:23:40 +08:00
..
__init__.py fix typo 2025-02-26 18:37:22 +08:00
gemm.py Fix pipeline stage edge cases 2025-05-07 11:40:34 +08:00
m_grouped_gemm.py Refactor JIT compilation (+NVRTC support) (#94) 2025-05-07 11:38:14 +08:00
runtime.py Fix 12.9 compatibility 2025-05-07 13:23:40 +08:00
tuner.py Add DG_PRINT_AUTOTUNE to README 2025-05-07 11:46:52 +08:00
utils.py Fix get_col_major_tma_aligned_tensor to handle 2-dimensional inputs 2025-03-13 22:15:16 +08:00