| # Copyright (c) Microsoft Corporation. | |
| # SPDX-License-Identifier: Apache-2.0 | |
| # DeepSpeed Team | |
| from deepspeed.utils.torch import required_torch_version | |
| def is_torch_elastic_compatible(): | |
| ''' | |
| Helper to lookup torch version. Elastic training is | |
| introduced in 1.11.x | |
| ''' | |
| return required_torch_version(min_version=1.11) | |