wehub-resource-sync
20 lines
477 B
Python
20 lines
477 B
Python
# 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'
|