-
-
Notifications
You must be signed in to change notification settings - Fork 10.8k
[Misc] Support more collective_rpc return types #25294
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
Signed-off-by: Nick Hill <nhill@redhat.com>
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.
Code Review
This pull request generalizes the serialization of collective_rpc return types to support arbitrarily nested lists and dictionaries, which is a valuable enhancement. The core logic is implemented through new recursive encoding and decoding functions in vllm/v1/serial_utils.py, and the changes are well-covered by new tests. My review includes suggestions to improve the robustness of the deserialization logic by adding assertions to guard against potential data mismatches, which could lead to silent errors.
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.
Thanks for taking the time to implement this!
Signed-off-by: Nick Hill <nhill@redhat.com>
Signed-off-by: Nick Hill <nhill@redhat.com> Signed-off-by: charlifu <charlifu@amd.com>
Signed-off-by: Nick Hill <nhill@redhat.com> Signed-off-by: yewentao256 <zhyanwentao@126.com>
Signed-off-by: Nick Hill <nhill@redhat.com> Signed-off-by: xuebwang-amd <xuebwang@amd.com>
Signed-off-by: Nick Hill <nhill@redhat.com>
Signed-off-by: Nick Hill <nhill@redhat.com>
Signed-off-by: Nick Hill <nhill@redhat.com> Signed-off-by: xuebwang-amd <xuebwang@amd.com>
Generalization of #21845, support arbitrarily nested combinations of dicts and lists in returned types.
This is needed for #18465.