fix bugs in forward_end

Author: qmkakaxi

vllm_ascend/eplb/eplb_updator.py

@@ -130,14 +130,17 @@ def forward_before(self):
         self.reqs = []
         self.eplb_loader.asyn_expert_weight_transfer(self.reqs)
 
-    def forward_end(self, dummy_run=False):
+def forward_end(self,dummy_run=False):
         self.adaptor.get_rank_expert_workload(self.num_moe_layers,dummy_run)
-        if not self.update_in_flight and self.get_update_iteration():
-            moe_load = self.compute_and_set_moe_load(dummy_run)
-            self.wakeup_eplb_worker()
-            self.update_in_flight = True
-            self.wait_worker_iterations = 0
-            self.weight_loading = False
+        if not self.update_in_flight:
+            load_gather_iteration, update_iteration = self.get_update_iteration()
+            if load_gather_iteration:
+                moe_load = self.compute_and_set_moe_load(dummy_run)
+            if update_iteration:
+                self.wakeup_eplb_worker()
+                self.update_in_flight = True
+                self.wait_worker_iterations = 0
+                self.weight_loading = False
 
         if self.update_in_flight:
             self.wait_worker_iterations = self.wait_worker_iterations + 1