TransformerEngine

1
# Copyright (c) 2022-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2
#
3
# See LICENSE for license information.
4

5
"""Version information"""
6
import sys
7
from packaging.version import Version
8

9
if sys.version_info >= (3, 8):
10
    from importlib import metadata
11
else:
12
    import importlib_metadata as metadata
13

14
def _version_str() -> str:
15
    """Transformer Engine version string"""
16

17
    # Try getting version from package metadata
18
    version_str = None
19
    try:
20
        version_str = metadata.version("transformer_engine")
21
    except:
22
        pass
23
    if version_str:
24
        return version_str
25

26
    # Try getting version from Git root directory
27
    try:
28
        from te_version import te_version
29
        version_str = te_version()
30
    except:
31
        pass
32
    if version_str:
33
        return version_str
34

35
    # Could not deduce version
36
    return "0.dev0+unknown"
37

38
# Transformer Engine version
39
__version__: Version = Version(_version_str())
40