DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-04-06 01:34:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix typo

Browse Source
This commit is contained in:
dxh 2025-03-07 11:42:51 +08:00
parent 0b5d353dba
commit de1603d5e4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deep_gemm/jit_kernels/config.py
View File

@ -86,12 +86,12 @@ class ConfigCache:
self.cached = {} self.cached = {}
def compute_and_cache(self, m: int, n: int, k: int, num_groups: int, num_sms: int, def compute_and_cache(self, m: int, n: int, k: int, num_groups: int, num_sms: int,
is_grouped_contigous: bool = False) -> Tuple[int, int, int, int]: is_grouped_contiguous: bool = False) -> Tuple[int, int, int, int]:
signature = str((m, n, k, num_groups, num_sms, is_grouped_contigous)) signature = str((m, n, k, num_groups, num_sms, is_grouped_contiguous))
signature = hash_to_hex(signature) signature = hash_to_hex(signature)
if signature in self.cached: if signature in self.cached:
return self.cached[signature] return self.cached[signature]
best_config = get_best_configs(m, n, k, num_groups, num_sms, is_grouped_contigous) best_config = get_best_configs(m, n, k, num_groups, num_sms, is_grouped_contiguous)
self.cached[signature] = best_config self.cached[signature] = best_config
return best_config return best_config