Skip to content

[RPC] Python UDF and Builtin Ops should serialize to the same message layout #24252

@mrshenli

Description

@mrshenli

As pointed out by @zdevito in #23569, instead of serializing the entire Python UDF RPC into a byte vector, it should stay consistent in the message layout with Builtin Ops where tensors are stored in a separate tensor table. This design avoids serializing everything all at once, and makes it possible to introduce streaming serialization in the future.

cc @pietern @mrshenli @pritamdamania87 @zhaojuanmao @satgera

Metadata

Metadata

Assignees

Labels

module: rpcRelated to RPC, distributed autograd, RRef, and distributed optimizeroncall: distributedAdd this issue/PR to distributed oncall triage queuetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions