Asynchronous Disk Access in Workers
- Dominant language
- Python
- Stars
- 1.7k
- Forks
- 778
- Avg merge
- 2h 50m
- Merged PRs (30d)
- 3
Description
Currently reading from or writing to disk blocks the event loop
https://github.com/dask/distributed/blob/74e8dc64ef0436147a88ba225604ed7f86b0d569/distributed/worker.py#L1948-L1964
This can cause workers to become unresponsive, especially in systems with very slow disk access. Ideally Disk I/O would happen concurrently. There are a couple of ways to do this.
## Offload to separate thread
We could move all manipulation of the `Worker.data` MutableMapping to a separate thread, such as we do with the `offload` function, which we use today for deserialization.
However, if we do this then we need to do it for *all* access to `Worker.data` including seemingly innocuous checks like `if key in self.data` which may become annoying.
## Handle Disk logic directly in the worker
We could also break apart the `MutableMapping` abstraction, and unpack the zict logic directly into the Worker code. This would allow us to keep a lot of the fast access in the event loop, while treating disk access specially. It would also open the door for more performance improvements, like trying to schedule tasks for data that is currently in memory rather than data that is currently on disk. In general if we want to improve out-of-memory handling in Dask we'll eventually need to break this abstraction.
However, breaking this abstraction comes at considerable cost. First, it means that there is more to manage in a monolithic Worker codebase (zict has tricky logic that we haven't really had to touch or maintain in years). Second, it means that we'll have to find a way that still lets other groups like RAPIDS extend the storage hierarchy (they have device->host->disk rather than just host->disk).
cc @quasiben @pentschev @jrbourbeau @fjetter
Contributor guide
Assessment
This issue has not been assessed yet.