Skip to content

Commit

Permalink
[worker] remove unused assertion (#161)
Browse files Browse the repository at this point in the history
### 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 <cmq0113@163.com>
  • Loading branch information
MengqingCao authored Feb 26, 2025
1 parent 78530c0 commit 1c238b9
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions vllm_ascend/worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 1c238b9

Please sign in to comment.