File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -1641,7 +1641,7 @@ def _dummy_run(
1641
1641
attn_state = attn_state ,
1642
1642
)
1643
1643
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 :
1645
1645
self .eplb_updator .forward_before ()
1646
1646
1647
1647
with self .maybe_dummy_run_with_lora (self .lora_config ,
@@ -1728,7 +1728,7 @@ def _dummy_run(
1728
1728
1729
1729
if is_profile_run and self .dynamic_eplb :
1730
1730
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 :
1732
1732
self .eplb_updator .take_update_info_from_eplb_process ()
1733
1733
self .eplb_updator .forward_end ()
1734
1734
You can’t perform that action at this time.
0 commit comments