fix logits_indices in _process_req

Signed-off-by: zzhx1 <zzh_201018@outlook.com>

Author: zzhx1

vllm_ascend/worker/model_runner_v1.py

@@ -1320,14 +1320,14 @@ def _process_reqs(
         if self.use_aux_hidden_state_outputs:
             hidden_states, aux_hidden_states = hidden_states
 
-        if _enable_lmhead_tp():  #
+        if _enable_lmhead_tp(): 
             if not with_prefill:
                 max_num_reqs_across_dp = padded_num_tokens_across_dp
             else:
                 max_num_reqs_across_dp = self.max_num_reqs
-            sample_indices = nn.functional.pad(
-                sample_indices,
-                (0, max_num_reqs_across_dp - sample_indices.shape[0]))
+            logits_indices = nn.functional.pad(
+                logits_indices,
+                (0, max_num_reqs_across_dp - logits_indices.shape[0]))
 
         return (attn_metadata, hidden_states, spec_decode_metadata, positions,
                 total_num_scheduled_tokens, logits_indices, aux_hidden_states,