constants.py 665 B

123456789101112131415161718192021222324252627
  1. """
  2. "Copyright 2020 The Microsoft DeepSpeed Team.
  3. Licensed under the MIT license.
  4. """
  5. #########################################
  6. # AIO
  7. #########################################
  8. AIO_FORMAT = '''
  9. "aio": {
  10. "block_size": 1048576,
  11. "queue_depth": 8,
  12. "thread_count": 1,
  13. "single_submit": false,
  14. "overlap_events": true
  15. }
  16. '''
  17. AIO = "aio"
  18. AIO_BLOCK_SIZE = "block_size"
  19. AIO_BLOCK_SIZE_DEFAULT = 1048576
  20. AIO_QUEUE_DEPTH = "queue_depth"
  21. AIO_QUEUE_DEPTH_DEFAULT = 8
  22. AIO_THREAD_COUNT = "thread_count"
  23. AIO_THREAD_COUNT_DEFAULT = 1
  24. AIO_SINGLE_SUBMIT = "single_submit"
  25. AIO_SINGLE_SUBMIT_DEFAULT = False
  26. AIO_OVERLAP_EVENTS = "overlap_events"
  27. AIO_OVERLAP_EVENTS_DEFAULT = True