DeepSpeed

1
# Copyright (c) Microsoft Corporation.
2
# SPDX-License-Identifier: Apache-2.0
3

4
# DeepSpeed Team
5

6
try:
7
    #  This is populated by setup.py
8
    from .git_version_info_installed import *  # noqa: F401 # type: ignore
9
except ModuleNotFoundError:
10
    import os
11
    if os.path.isfile('version.txt'):
12
        # Will be missing from checkouts that haven't been installed (e.g., readthedocs)
13
        version = open('version.txt', 'r').read().strip()
14
    else:
15
        version = "0.0.0"
16
    git_hash = '[none]'
17
    git_branch = '[none]'
18

19
    from .ops.op_builder.all_ops import ALL_OPS
20
    installed_ops = dict.fromkeys(ALL_OPS.keys(), False)
21
    accelerator_name = ""
22
    torch_info = {'version': "0.0", "cuda_version": "0.0", "hip_version": "0.0"}
23

24
# compatible_ops list is recreated for each launch
25
from .ops.op_builder.all_ops import ALL_OPS
26

27
compatible_ops = dict.fromkeys(ALL_OPS.keys(), False)
28
for op_name, builder in ALL_OPS.items():
29
    op_compatible = builder.is_compatible()
30
    compatible_ops[op_name] = op_compatible
31
    compatible_ops["deepspeed_not_implemented"] = False
32