Skip to content

Commit b39d2f4

Browse files
committed
graceful cleanup
Signed-off-by: zhihuiwan <15779896112@163.com>
1 parent a83ee48 commit b39d2f4

1 file changed

Lines changed: 9 additions & 6 deletions

File tree

python/fate_flow/controller/task_controller.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -207,12 +207,15 @@ def kill_task(cls, task: Task):
207207
kill_status = False
208208
try:
209209
# kill task executor
210-
backend_engine = build_engine(
211-
task.f_engine_conf.get("computing_engine"),
212-
task.f_is_deepspeed
213-
)
214-
if backend_engine:
215-
backend_engine.kill(task)
210+
try:
211+
backend_engine = build_engine(
212+
task.f_engine_conf.get("computing_engine"),
213+
task.f_is_deepspeed
214+
)
215+
if backend_engine:
216+
backend_engine.kill(task)
217+
except Exception as e:
218+
schedule_logger(task.f_job_id).exception(e)
216219
WorkerManager.kill_task_all_workers(task)
217220
except Exception as e:
218221
schedule_logger(task.f_job_id).exception(e)

0 commit comments

Comments
 (0)