EngineX/xc-llm-kunlun
3
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

[chore] Remove obsolete comments

Browse Source
This commit is contained in:
zhaoyingzhuo
2025-12-10 15:52:23 +08:00
parent ec935627cb
commit b614823125
6 changed files with 1 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
vllm_kunlun/ops/sample/sampler.py
View File

@@ -516,10 +516,6 @@ def _apply_top_k_top_p(
top_p_mask[:, -1] = False
logits_sort.masked_fill_(top_p_mask, -float("inf"))
# Re-sort the probabilities.
# logits = torch.empty_like(logits_sort).scatter_(dim=-1,
# index=logits_idx,
# src=logits_sort)
return logits_sort, logits_idx
@@ -883,7 +879,6 @@ def _sample_with_torch(
seq_groups=seq_groups_arg,
)
if logits_idx is not None:
# multinomial_samples[sampling_type] = logits_idx[:, result_idx[:][0]]
token_ids = logits_idx[long_sample_indices].gather(
dim=1, index=result_idx.to(logits_idx.device)
)