validate_async_io.py 263 Bytes
Newer Older
aiss's avatar
aiss committed
1
2
3
4
5
6
7
8
9
"""
Copyright 2021 The Microsoft DeepSpeed Team
Licensed under the MIT license.

Functionality of swapping optimizer tensors to/from (NVMe) storage devices.
"""
import deepspeed
from deepspeed.ops.aio import AsyncIOBuilder
assert AsyncIOBuilder().is_compatible()