| # Copyright (c) Microsoft Corporation. | |
| # SPDX-License-Identifier: Apache-2.0 | |
| # DeepSpeed Team | |
| AIO_HANDLE = 'aio_handle' | |
| AIO_BASIC = 'aio_basic' | |
| TORCH_IO = 'torch_io' | |
| TORCH_FAST_IO = 'torch_fastio' | |
| VALID_ENGINES = [AIO_HANDLE, AIO_BASIC, TORCH_IO, TORCH_FAST_IO] | |
| BUFFER = 'buffer' | |
| BOUNCE_BUFFER = 'bounce_buffer' | |
| NUM_BYTES = 'num_bytes' | |
| FILE = 'file' | |
| HANDLE = 'handle' | |
| ELAPSED_SEC = 'elapsed_sec' | |
| FAST_IO_BUFFER = 'fast_io_buffer' | |
| USE_CPU_LOCKED_TENSOR = 'cpu_locked_tensor' | |
| USE_GDS = 'gds' | |