| # Copyright (c) Microsoft Corporation. | |
| # SPDX-License-Identifier: Apache-2.0 | |
| # DeepSpeed Team | |
| """ | |
| Functionality of swapping optimizer tensors to/from (NVMe) storage devices. | |
| """ | |
| from deepspeed.ops.op_builder import AsyncIOBuilder | |
| assert AsyncIOBuilder().is_compatible() | |
| assert AsyncIOBuilder().load() | |