refine some typo (#3473)

This commit is contained in:
Xiaoyu Zhang
2025-02-10 23:35:44 +08:00
committed by GitHub
parent 4fe92bfca5
commit 2f47d710ae
4 changed files with 4 additions and 4 deletions

View File

@@ -75,7 +75,7 @@ def fused_topk(
return topk_weights, topk_ids
# This is used by the Deepseek-V2 model
# This is used by the Deepseek V2/V3/R1 series models
@torch.compile(dynamic=True, backend=get_compiler_backend())
def grouped_topk(
hidden_states: torch.Tensor,