minor

Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>

vllm/v1/worker/gpu/sampler.py

@@ -254,8 +254,8 @@ def compute_logprobs(
     )
 
     # NOTE(woosuk): Here, to save GPU memory, we do not materialize the full
-    # logprobs tensor. Instead, we only compute the logprobs of the topk + 1
-    # tokens.
+    # logprobs tensor. Instead, we only compute and return the logprobs of
+    # the topk + 1 tokens.
     BLOCK_SIZE = 1024
     _topk_logprobs_kernel[(batch_size, )](
         logprobs,