Skip to content

Commit c3fc14f

Browse files
bug fix
1 parent 4050d14 commit c3fc14f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm_ascend/worker/model_runner_v1.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1641,7 +1641,7 @@ def _dummy_run(
16411641
attn_state=attn_state,
16421642
)
16431643

1644-
if not is_compile and not is_profile_run and self.dynamic_eplb:
1644+
if not is_torchair_compile and not is_profile_run and self.dynamic_eplb:
16451645
self.eplb_updator.forward_before()
16461646

16471647
with self.maybe_dummy_run_with_lora(self.lora_config,
@@ -1728,7 +1728,7 @@ def _dummy_run(
17281728

17291729
if is_profile_run and self.dynamic_eplb:
17301730
self.model.clear_all_moe_loads()
1731-
if not is_compile and not is_profile_run and self.dynamic_eplb:
1731+
if not is_torchair_compile and not is_profile_run and self.dynamic_eplb:
17321732
self.eplb_updator.take_update_info_from_eplb_process()
17331733
self.eplb_updator.forward_end()
17341734

0 commit comments

Comments
 (0)