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

Fix typing

Browse Source
This commit is contained in:
Chenggang Zhao 2025-05-07 11:22:02 +08:00
parent c5fa9f1068
commit a29b331c48
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deep_gemm/jit/runtime.py
View File

@ -73,7 +73,7 @@ class RuntimeCache:
def __init__(self) -> None:
self.cache = {}
def __setitem__(self, path, runtime) -> None:
def __setitem__(self, path: str, runtime: Runtime) -> None:
self.cache[path] = runtime
def get(self, path: str, runtime_cls: Type[Runtime]) -> Optional[Runtime]: