jlewi / flaap

Federated Learning and Analytics Protocols
Apache License 2.0
0 stars 0 forks source link

Make worker stateful - #22 #25

Closed jlewi closed 2 years ago

jlewi commented 2 years ago

Per #22 we need to make the workers stateful.

This PR changes the protocol so that we no longer fuse the operations to send the values to the worker to compute and the actual invocation of the function in the remote worker.

This requires redesigning the Task proto so that we can send the different types of requests (create_value, create_call, compute, etc...). Results are only materialized in the task in response to a compute task.

We also add a group_index field to Tasks. This is used to track the order in which tasks are created by the executor and to ensure they are processed in the order they are created.