Skip to content

[HashShuffle] - Add warnings for when there are insufficient resources for Aggregators #53705

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Conversation

goutamvenkat-anyscale
Copy link
Contributor

@goutamvenkat-anyscale goutamvenkat-anyscale commented Jun 10, 2025

Why are these changes needed?

Improve the observability for hash based shuffling.

What is the change?

  1. At the start of executing HashShufflingOperatorBase, check if the cluster has enough resources to schedule all actors, if not we can fail outright.

  2. Add a warning to continue observing if all of the Aggregators are up and running after a period of 30 secs (configurable), if not print a warning.

Checks

  • I've signed off every commit(by using the -s flag, i.e., git commit -s) in this PR.
  • I've run scripts/format.sh to lint the changes in this PR.
  • I've included any doc changes needed for https://docs.ray.io/en/master/.
    • I've added any new APIs to the API Reference. For example, if I added a
      method in Tune, I've added it in doc/source/tune/api/ under the
      corresponding .rst file.
  • I've made sure the tests are passing. Note that there might be a few flaky tests, see the recent failures at https://flakey-tests.ray.io/
  • Testing Strategy
    • Unit tests
    • Release tests
    • This PR is not tested :(

@goutamvenkat-anyscale goutamvenkat-anyscale requested a review from a team as a code owner June 10, 2025 17:32
@goutamvenkat-anyscale goutamvenkat-anyscale force-pushed the goutam/hash_shuffle_logging branch 2 times, most recently from 5f03c85 to 935f5ec Compare June 10, 2025 17:42
goutamvenkat-anyscale and others added 8 commits June 16, 2025 12:04
…s for Aggregators

Signed-off-by: goutam <[email protected]>
Signed-off-by: Goutam V <[email protected]>
Signed-off-by: Goutam V <[email protected]>
Signed-off-by: Goutam V <[email protected]>
Signed-off-by: Goutam V <[email protected]>
Signed-off-by: Goutam V <[email protected]>
Signed-off-by: Goutam V <[email protected]>
@goutamvenkat-anyscale goutamvenkat-anyscale force-pushed the goutam/hash_shuffle_logging branch from 03232e7 to 83bb03b Compare June 16, 2025 19:04
@goutamvenkat-anyscale goutamvenkat-anyscale added go add ONLY when ready to merge, run all tests and removed go add ONLY when ready to merge, run all tests labels Jun 16, 2025
Signed-off-by: Goutam V <[email protected]>
@goutamvenkat-anyscale goutamvenkat-anyscale added go add ONLY when ready to merge, run all tests and removed go add ONLY when ready to merge, run all tests labels Jun 17, 2025
Signed-off-by: Goutam V <[email protected]>
Signed-off-by: Goutam V <[email protected]>
@goutamvenkat-anyscale goutamvenkat-anyscale added the go add ONLY when ready to merge, run all tests label Jun 17, 2025
Signed-off-by: Goutam V <[email protected]>
Signed-off-by: Goutam V <[email protected]>
@alexeykudinkin alexeykudinkin enabled auto-merge (squash) June 18, 2025 02:23
@alexeykudinkin
Copy link
Contributor

@goutamvenkat-anyscale enabled auto-merge please update the message before merging

@alexeykudinkin alexeykudinkin disabled auto-merge June 18, 2025 03:25
Signed-off-by: Goutam V <[email protected]>
@alexeykudinkin alexeykudinkin merged commit 9e37790 into ray-project:master Jun 19, 2025
5 checks passed
minerharry pushed a commit to minerharry/ray that referenced this pull request Jun 27, 2025
…s for Aggregators (ray-project#53705)

<!-- Thank you for your contribution! Please review
https://github.com/ray-project/ray/blob/master/CONTRIBUTING.rst before
opening a pull request. -->

<!-- Please add a reviewer to the assignee section when you create a PR.
If you don't have the access to it, we will shortly find a reviewer and
assign them to your PR. -->

## Why are these changes needed?
Improve the observability for hash based shuffling.

## What is the change?

1. At the start of executing HashShufflingOperatorBase, check if the
cluster has enough resources to schedule all actors, if not we can fail
outright.

2. Add a warning to continue observing if all of the Aggregators are up
and running after a period of 30 secs (configurable), if not print a
warning.

## Checks

- [x] I've signed off every commit(by using the -s flag, i.e., `git
commit -s`) in this PR.
- [x] I've run `scripts/format.sh` to lint the changes in this PR.
- [ ] I've included any doc changes needed for
https://docs.ray.io/en/master/.
- [ ] I've added any new APIs to the API Reference. For example, if I
added a
method in Tune, I've added it in `doc/source/tune/api/` under the
           corresponding `.rst` file.
- [ ] I've made sure the tests are passing. Note that there might be a
few flaky tests, see the recent failures at https://flakey-tests.ray.io/
- Testing Strategy
   - [ ] Unit tests
   - [ ] Release tests
   - [ ] This PR is not tested :(

---------

Signed-off-by: goutam <[email protected]>
Signed-off-by: Goutam V <[email protected]>
Co-authored-by: Goutam Venkatramanan <[email protected]>
elliot-barn pushed a commit that referenced this pull request Jul 2, 2025
…s for Aggregators (#53705)

<!-- Thank you for your contribution! Please review
https://github.com/ray-project/ray/blob/master/CONTRIBUTING.rst before
opening a pull request. -->

<!-- Please add a reviewer to the assignee section when you create a PR.
If you don't have the access to it, we will shortly find a reviewer and
assign them to your PR. -->

## Why are these changes needed?
Improve the observability for hash based shuffling.

## What is the change?

1. At the start of executing HashShufflingOperatorBase, check if the
cluster has enough resources to schedule all actors, if not we can fail
outright.

2. Add a warning to continue observing if all of the Aggregators are up
and running after a period of 30 secs (configurable), if not print a
warning.

## Checks

- [x] I've signed off every commit(by using the -s flag, i.e., `git
commit -s`) in this PR.
- [x] I've run `scripts/format.sh` to lint the changes in this PR.
- [ ] I've included any doc changes needed for
https://docs.ray.io/en/master/.
- [ ] I've added any new APIs to the API Reference. For example, if I
added a
method in Tune, I've added it in `doc/source/tune/api/` under the
           corresponding `.rst` file.
- [ ] I've made sure the tests are passing. Note that there might be a
few flaky tests, see the recent failures at https://flakey-tests.ray.io/
- Testing Strategy
   - [ ] Unit tests
   - [ ] Release tests
   - [ ] This PR is not tested :(

---------

Signed-off-by: goutam <[email protected]>
Signed-off-by: Goutam V <[email protected]>
Co-authored-by: Goutam Venkatramanan <[email protected]>
Signed-off-by: elliot-barn <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
go add ONLY when ready to merge, run all tests
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants