reduce_op.py 259 B

123456789101112131415161718
  1. # Copyright (c) Microsoft Corporation.
  2. # SPDX-License-Identifier: Apache-2.0
  3. # DeepSpeed Team
  4. from enum import Enum
  5. class ReduceOp(Enum):
  6. SUM = 0
  7. PRODUCT = 1
  8. MIN = 2
  9. MAX = 3
  10. BAND = 4
  11. BOR = 5
  12. BXOR = 6
  13. AVG = 7
  14. UNUSED = 8