dask / dask/distributed

Monitor spilled data that is still referenced

Open
#6,220 0 comments 0 reactions 0 assignees View on GitHub
memory
Dominant language
Python
Stars
1.7k
Forks
778
Avg merge
2h 50m
Merged PRs (30d)
3

Description

Follow-up to #5936

In a normal situation, a worker holds some in-memory tasks that are relatively at rest while it is actively working on others.
In that case, in case of memory pressure, the tasks at rest are spilled to disk using a LRU algorithm.

In case of extreme memory pressure, however, a key may be spilled to disk while it is in use - either because it's an input to another task that's currently running or because it's being sent to another worker. When that happens, the data is spilled but the memory is not released until the compute or send has finished; in the GUI, its RAM will transition from "managed" to a double effect of "unmanaged recent" plus "spilled".

It would be valuable to separate this kind of memory usage from the opaque unmanaged blob.
This is straightforward after #5936:

```python
class SpillBuffer:
@property
def spilled_but_still_referenced(self) -> int:
if not has_zict_220:
return 0
cache = cast(Cache, self.slow)
slow = cast(Slow, cache.data)
return sum(slow.weight_by_key[key].memory for key in cache.cache)
```
The above is O(n) to the number of active computations and transfers - so negligible most times.
The output could be sent to the scheduler during heartbeat and contribute to ``distributed.scheduler.MemoryState``, like it already happens for ``SpillBuffer.spilled_total``.

#### TODO
Come up with a good way to visualize this info in the GUI
#### OUT OF SCOPE
UIse the new metric in algorithms (but feel free to discuss here)

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.