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

feat: drop support for CUDA<12.3

Browse Source 
Signed-off-by: Zihua Wu <13583761+lucifer1004@users.noreply.github.com>
This commit is contained in:
Zihua Wu
2025-04-25 18:56:40 -07:00
parent 46762b6903
commit f6198492cb
5 changed files with 46 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
deep_gemm/jit/template.py
View File

@@ -24,7 +24,6 @@ def generate(**kwargs: Dict[str, Any]) -> str:
using namespace deep_gemm;
#ifndef NVRTC_JIT_COMPILATION
__global__ void dummy_kernel() {{
void *ptr = (void *)&fp8_gemm_kernel<
{kwargs['N']},
@@ -43,7 +42,6 @@ __global__ void dummy_kernel() {{
GemmType::{kwargs['GEMM_TYPE']}
>;
}}
#endif
'''
# Debug print