Pre-stored version information

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

Author: lianyiibo

vllm_ascend/core/scheduler.py

@@ -53,6 +53,7 @@ def __init__(
                          include_finished_set, log_stats)
         self.scheduled_req_ids: set[str] = set()
         self.running: list[Request] = []
+        self.lock_version = vllm_version_is("0.9.2")
 
     def schedule(self) -> SchedulerOutput:
         if self.scheduler_config.chunked_prefill_enabled:
@@ -283,13 +284,13 @@ def skip_cur_request():
                     # allow the lower-priority requests to be scheduled.
                     req_index += 1
                     continue
-                if vllm_version_is("0.9.2"):
+                if self.lock_version:
                     num_draft_tokens = max(
                         num_new_tokens + request.num_computed_tokens -
                         request.num_tokens, 0)
 
                 while True:
-                    if vllm_version_is("0.9.2"):
+                    if self.lock_version:
                         new_blocks = self.kv_cache_manager.allocate_slots(
                             request,
                             num_new_tokens,