Skip to content

Commit 89dd326

Browse files
Merge commit 'refs/pull/4741/head' of https://github.com/vllm-project/vllm-ascend into eplb_mix
2 parents f93bc2a + bc1cb89 commit 89dd326

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vllm_ascend/eplb/core/policy/policy_factory.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from .policy_abstract import DynamicConfig, EplbPolicy
44
from .policy_dynamic_ep import DynamicEplb
55
from .policy_dynamic_ep_v2 import DynamicEplbV2
6-
from .policy_flashlb import FlashLB
6+
from .policy_flashlb import FlashLB, warm_up
77
from .policy_random import RandomLoadBalance
88

99

@@ -29,5 +29,5 @@ def generate_policy(policy_type: int, config: DynamicConfig) -> EplbPolicy:
2929
policy_class = policy.get(policy_type, RandomLoadBalance)
3030
policy_instance = policy_class(config)
3131
if policy_type == 3:
32-
policy_instance.warm_up()
33-
return policy_instance
32+
warm_up()
33+
return policy_instance

0 commit comments

Comments
 (0)