DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-06-26 23:15:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

tested

Browse Source
This commit is contained in:
kavioyu
2025-03-12 13:48:02 +00:00
parent 9d3222a93e
commit 6e53c6613d
6 changed files with 678 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
deep_gemm/jit_kernels/__init__.py
View File

@@ -1,4 +1,5 @@
from .gemm import gemm_fp8_fp8_bf16_nt
from .gemm_bw import gemm_fp8_fp8_bf16_bw_nt
from .m_grouped_gemm import (
m_grouped_gemm_fp8_fp8_bf16_nt_contiguous,
m_grouped_gemm_fp8_fp8_bf16_nt_masked