From 1c238b930d2b21a37140a1b32e5ac12465fe5c6f Mon Sep 17 00:00:00 2001 From: Mengqing Cao Date: Wed, 26 Feb 2025 16:11:36 +0800 Subject: [PATCH] [worker] remove unused assertion (#161) ### What this PR does / why we need it? Remove unused assertion in `NPUWorker`, as this has been moved to `Executor` in vLLM: https://github.com/vllm-project/vllm/blob/aabeb2688fba861a39547a2a33649e6330caaafd/vllm/executor/uniproc_executor.py#L43 ### Does this PR introduce _any_ user-facing change? N/A ### How was this patch tested? CI passed with existing test. Signed-off-by: MengqingCao --- vllm_ascend/worker.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/vllm_ascend/worker.py b/vllm_ascend/worker.py index 27930d4d..d1e0765a 100644 --- a/vllm_ascend/worker.py +++ b/vllm_ascend/worker.py @@ -84,9 +84,6 @@ def __init__( self.distributed_init_method = distributed_init_method self.is_driver_worker = is_driver_worker - if is_driver_worker: - assert rank % self.parallel_config.tensor_parallel_size == 0, \ - "Driver worker should be rank 0 of tensor parallel group." if self.model_config.trust_remote_code: # note: lazy import to avoid importing torch before initializing from vllm.utils import init_cached_hf_modules