Skip to content

GrpcRuntime: only send arguments to annotated placements #1132

@mortendahl

Description

@mortendahl

In the eDSL we have the placement where an argument is coming from:

def simple_computation(
    x: pm.Argument(placement=alice, vtype=pm.TensorType(pm.float64)),
    y: pm.Argument(placement=bob, vtype=pm.TensorType(pm.float64)),
):

which means we might be able to only send x to Alice, and only y to Bob, instead of currently sending everything to everyone.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions