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/__init__.py
View File

@@ -3,6 +3,7 @@ import torch
from . import jit
from .jit_kernels import (
gemm_fp8_fp8_bf16_nt,
gemm_fp8_fp8_bf16_bw_nt,
m_grouped_gemm_fp8_fp8_bf16_nt_contiguous,
m_grouped_gemm_fp8_fp8_bf16_nt_masked,
ceil_div,