Skip to content

Commit eb92e7d

Browse files
authored
[Bugfix] Restore balance scheduling patch for v0.17.0 (vllm-project#7479)
### What this PR does / why we need it? Restore previously introduced patches: - vllm-project#5212 - vLLM version: v0.17.0 - vLLM main: vllm-project/vllm@8b63257 Signed-off-by: MrZ20 <2609716663@qq.com>
1 parent 95e1dc1 commit eb92e7d

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

vllm_ascend/patch/platform/__init__.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@
1919
import vllm_ascend.patch.platform.patch_distributed # noqa
2020
import vllm_ascend.patch.platform.patch_fusion_matcher_compat_ops # noqa
2121
import vllm_ascend.patch.platform.patch_kv_cache_interface # noqa
22-
from vllm_ascend.utils import is_310p
22+
from vllm_ascend import envs
23+
from vllm_ascend.utils import is_310p, vllm_version_is
2324

2425
if not is_310p():
2526
import vllm_ascend.patch.platform.patch_mamba_config # noqa
@@ -31,3 +32,5 @@
3132

3233
if os.getenv("DYNAMIC_EPLB", "false").lower() in ("true", "1") or os.getenv("EXPERT_MAP_RECORD", "false") == "true":
3334
import vllm_ascend.patch.platform.patch_multiproc_executor # noqa
35+
if envs.VLLM_ASCEND_BALANCE_SCHEDULING and vllm_version_is("0.17.0"):
36+
import vllm_ascend.patch.platform.patch_balance_schedule # noqa

0 commit comments

Comments
 (0)