constants.py 596 B

123456789101112131415161718192021222324252627
  1. # Copyright (c) Microsoft Corporation.
  2. # SPDX-License-Identifier: Apache-2.0
  3. # DeepSpeed Team
  4. """
  5. AIO
  6. """
  7. AIO_FORMAT = '''
  8. "aio": {
  9. "block_size": 1048576,
  10. "queue_depth": 8,
  11. "thread_count": 1,
  12. "single_submit": false,
  13. "overlap_events": true
  14. }
  15. '''
  16. AIO = "aio"
  17. AIO_BLOCK_SIZE = "block_size"
  18. AIO_BLOCK_SIZE_DEFAULT = 1048576
  19. AIO_QUEUE_DEPTH = "queue_depth"
  20. AIO_QUEUE_DEPTH_DEFAULT = 8
  21. AIO_THREAD_COUNT = "thread_count"
  22. AIO_THREAD_COUNT_DEFAULT = 1
  23. AIO_SINGLE_SUBMIT = "single_submit"
  24. AIO_SINGLE_SUBMIT_DEFAULT = False
  25. AIO_OVERLAP_EVENTS = "overlap_events"
  26. AIO_OVERLAP_EVENTS_DEFAULT = True