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/python-package/xgboost/collective.py
+++ b/python-package/xgboost/collective.py
@@ -191,6 +191,9 @@ class Op(IntEnum):
    MAX = 0
    MIN = 1
    SUM = 2
+    BITWISE_AND = 3
+    BITWISE_OR = 4
+    BITWISE_XOR = 5


 def allreduce(  # pylint:disable=invalid-name