fix pin_memory error

Signed-off-by: Ronald1995 <[email protected]>

Author: Ronald1995

vllm_ascend/worker/model_runner_v1.py

@@ -2122,7 +2122,7 @@ def _calc_spec_decode_metadata(
                 cu_num_scheduled_tokens - num_sampled_tokens,
                 num_sampled_tokens)
             logits_indices_pcp += arange
-            logits_indices_pcp = torch.tensor(logits_indices_pcp, pin_memory=True).to(
+            logits_indices_pcp = torch.from_numpy(logits_indices_pcp).pin_memory().to(
                 self.device, non_blocking=True)
 
         # Compute the bonus logits indices.
@@ -2145,25 +2145,27 @@ def _calc_spec_decode_metadata(
 
         # TODO: Optimize the CPU -> NPU copy.
         cu_num_draft_tokens = (
-            torch.tensor(cu_num_draft_tokens, pin_memory=True)
+            torch.from_numpy(cu_num_draft_tokens)
+            .pin_memory()
             .to(self.device, non_blocking=True)
         )
         cu_num_sampled_tokens = (
-            torch.tensor(cu_num_sampled_tokens, pin_memory=True)
+            torch.from_numpy(cu_num_sampled_tokens)
+            .pin_memory()
             .to(self.device, non_blocking=True)
         )
         logits_indices = (
-            torch.tensor(logits_indices, pin_memory=True)
+            torch.from_numpy(logits_indices)
+            .pin_memory()
             .to(self.device, non_blocking=True)
         )
         target_logits_indices = (
-            torch.tensor(target_logits_indices, pin_memory=True)
-            .to(self.device, non_blocking=True)
-        )
-        bonus_logits_indices = (
-            torch.tensor(bonus_logits_indices, pin_memory=True)
+            torch.from_numpy(target_logits_indices)
+            .pin_memory()
             .to(self.device, non_blocking=True)
         )
+        bonus_logits_indices = torch.from_numpy(bonus_logits_indices).pin_memory().to(
+            self.device, non_blocking=True)
 
         # Compute the draft token ids.
         # draft_token_indices:      [  1,   2,   3, 105, 106, 208]