File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -422,10 +422,10 @@ def only_decode(self):
422422 if self .fd_config .parallel_config .use_ep and self .fd_config .scheduler_config .splitwise_role == "mixed" :
423423 # 在ep场景下no_need_stop如果都是false,表示全部卡空闲,返回false,走高吞吐分支,否则为部分卡空闲,需要进一步判断
424424 no_need_stop_list = []
425- no_need_stops = self .not_need_stop ()
426- paddle .distributed .all_gather_object (no_need_stop_list , not no_need_stops )
427- no_need_stop = all (no_need_stop_list )
428- if no_need_stop :
425+ no_need_stop = self .not_need_stop ()
426+ paddle .distributed .all_gather_object (no_need_stop_list , not no_need_stop )
427+ if_all_device_empty = all (no_need_stop_list )
428+ if if_all_device_empty :
429429 if_only_decode = False
430430 else :
431431 only_decode_batch_list = []
You can’t perform that action at this time.
0 commit comments