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
[Core] Update doc for Ray core accelerator support #41849
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just a small grammar nit.
@@ -87,14 +87,14 @@ def construct_metadata() -> WorkerMetadata: | |||
node_id = ray.get_runtime_context().get_node_id() | |||
node_ip = ray.util.get_node_ip_address() | |||
hostname = socket.gethostname() | |||
resource_ids = ray.get_runtime_context().get_resource_ids() | |||
accelerator_ids = ray.get_runtime_context().get_accelerator_ids() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@matthewdeng I deprecated get_resource_ids() and introduced get_accelerator_ids() as we discussed. You may want to change train side as well.
- Include docs for accelerators other than GPU - Replace RuntimeContext.get_resource_ids() with RuntimeContext.get_accelerator_ids() Signed-off-by: Jiajun Yao <jeromeyjj@gmail.com>
Why are these changes needed?
Related issue number
Checks
git commit -s
) in this PR.scripts/format.sh
to lint the changes in this PR.method in Tune, I've added it in
doc/source/tune/api/
under thecorresponding
.rst
file.