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
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: def __init__(self) -> None:
self.cache = {} self.cache = {}
def __setitem__(self, path, runtime) -> None: def __setitem__(self, path: str, runtime: Runtime) -> None:
self.cache[path] = runtime self.cache[path] = runtime
def get(self, path: str, runtime_cls: Type[Runtime]) -> Optional[Runtime]: def get(self, path: str, runtime_cls: Type[Runtime]) -> Optional[Runtime]: