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

Functionality of swapping optimizer tensors to/from (NVMe) storage devices.
"""
aiss's avatar
aiss committed
7
from deepspeed.ops.op_builder import AsyncIOBuilder
aiss's avatar
aiss committed
8
assert AsyncIOBuilder().is_compatible()