Skip to content

Conversation

tgaddair
Copy link
Contributor

@tgaddair tgaddair commented Feb 5, 2021

Why are these changes needed?

wait_for_gpus does not handle the case when the user passes in an ordinal GPU ID as a string, which is the format returned by ray.get_gpu_ids(). As such, the default behavior of this function when gpu_id=None is currently broken.

Related issue number

#13486.

Checks

  • 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 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 :(

cc @richardliaw

@richardliaw richardliaw changed the title Fixed wait_for_gpu to handle str representations of ordinal IDs [tune] Fixed wait_for_gpu to handle str representations of ordinal IDs Feb 5, 2021
Signed-off-by: Richard Liaw <rliaw@berkeley.edu>
Signed-off-by: Richard Liaw <rliaw@berkeley.edu>
Copy link
Contributor

@richardliaw richardliaw left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice!

Signed-off-by: Richard Liaw <rliaw@berkeley.edu>
@richardliaw richardliaw merged commit cbd3598 into ray-project:master Feb 5, 2021
@tgaddair tgaddair deleted the fix-wait-for-gpus branch February 8, 2021 22:50
fishbone pushed a commit to fishbone/ray that referenced this pull request Feb 16, 2021
fishbone added a commit to fishbone/ray that referenced this pull request Feb 16, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants