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

Unify kwargs usages

Browse Source
This commit is contained in:
Chenggang Zhao
2025-05-15 16:53:52 +08:00
parent 350989eef3
commit 3b412f458a
6 changed files with 14 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
deep_gemm/jit_kernels/runtime.py
View File

@@ -138,7 +138,7 @@ class FP8GemmRuntime(Runtime):
super().__init__(path)
@staticmethod
def generate(**kwargs) -> str:
def generate(kwargs: Dict[str, Any]) -> str:
code = f'''
#ifdef __CUDACC_RTC__
#include <deep_gemm/nvrtc_std.cuh>
@@ -233,7 +233,7 @@ class FP8WGradGemmRuntime(Runtime):
super().__init__(path)
@staticmethod
def generate(**kwargs) -> str:
def generate(kwargs: Dict[str, Any]) -> str:
code = f'''
#ifdef __CUDACC_RTC__
#include <deep_gemm/nvrtc_std.cuh>