-
-
Notifications
You must be signed in to change notification settings - Fork 757
Description
This issue is a follow-up on an offline discussion centering around #6884 (comment) and #6939 with @crusaderky and @gjoseph92 where we encountered some issues with the way we currently handle tasks failing/succeeding on unexpected workers.
In transition_processing_memory, when encountering an unexpected worker that has successfully completed a task, we send a cancel-compute message to the worker that was supposed to process this task assuming that the unexpected worker would hold onto our result for us.
distributed/distributed/scheduler.py
Lines 1987 to 2001 in c15a10e
| if ws != ts.processing_on: # someone else has this task | |
| logger.info( | |
| "Unexpected worker completed task. Expected: %s, Got: %s, Key: %s", | |
| ts.processing_on, | |
| ws, | |
| key, | |
| ) | |
| assert ts.processing_on | |
| worker_msgs[ts.processing_on.address] = [ | |
| { | |
| "op": "cancel-compute", | |
| "key": key, | |
| "stimulus_id": stimulus_id, | |
| } | |
| ] |
However, this series of events should only be triggered in a chain of events where free-keys message for that task should already be on its way to the unexpected worker, causing it to remove the result. This leaves us with no worker holding on to the task/result.
While we do suspect this chain of events to unfold, we need a test verifying it. If this is indeed the case, we should not send the cancel-compute message to ts.processing_on, but rather make sure that we clean up anything related to the work happening on the unexpected worker.
In stimulus_task_erred, we don't check whether the task erred on an unexpected worker at all and run through the retry-logic. In #6884 (comment), @fjetter suggested instead ignoring that the task erred on an unexpected worker and continuing on as if nothing happened. This general approach is reasonable as the unexpected worker suggests something faulty going on. We might need to do additional work to ensure that the unexpected worker is reset, e.g., sending a free-keys message to the unexpected workers and we want to log a warning. To ensure everything is reset correctly, we need a test that verifies the intended behavior.