11 lines
302 B
Python
11 lines
302 B
Python
# 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()
|