[Data] Display logical memory in progress bar#63379
Merged
Merged
Conversation
The progress bar already shows CPUs, GPUs, and object store memory, but not logical memory — even though ExecutionResources tracks it and operators report it. Add logical memory to both the global resource status line and per-operator usage strings, conditional on non-zero usage. Signed-off-by: Balaji Veeramani <bveeramani@berkeley.edu>
Signed-off-by: Balaji Veeramani <bveeramani@berkeley.edu>
![gemini-code-assist[bot]](https://avatars.githubusercontent.com/in/956858?s=60&v=4){kind=small}
Contributor
There was a problem hiding this comment.
Code Review
This pull request integrates memory usage reporting into the Ray Data execution framework. It updates ResourceManager to include memory metrics in operator usage, allocations, and budgets, and refactors StreamingExecutor to include memory in its resource status reports. No review comments were provided, so I have no feedback to offer.
Signed-off-by: Balaji Veeramani <bveeramani@berkeley.edu>
Signed-off-by: Balaji Veeramani <bveeramani@berkeley.edu>
iamjustinhsu
approved these changes
May 15, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Description
The Ray Data progress bar shows CPU, GPU, and object store memory usage, but not logical memory — even though
ExecutionResourcesalready tracks it and operators already report it viacurrent_logical_usage(). This makes it hard to monitor memory-constrained workloads. This PR adds logical memory to both the global resource status line and per-operator usage strings, displayed conditionally when usage is non-zero (matching the existing GPU pattern).Related issues
None.
Additional information
The display order is: CPU, memory, GPU, object store. Memory is only shown when usage > 0, consistent with how GPU is handled.
Example global status line: