Support bitwise allreduce operations in the communicator (#8623)

2022-12-24 14:40:05 -08:00
parent c7e82b5914
commit 77b069c25d
10 changed files with 207 additions and 26 deletions
--- a/plugin/federated/federated.proto
+++ b/plugin/federated/federated.proto
@@ -25,6 +25,9 @@ enum ReduceOperation {
  MAX = 0;
  MIN = 1;
  SUM = 2;
+  BITWISE_AND = 3;
+  BITWISE_OR = 4;
+  BITWISE_XOR = 5;
 }

 message AllreduceRequest {