diff --git a/.gitignore b/.gitignore index 7356d0280033a6aebc494078a5f0f486efa1c3b0..9afa54e67d3fd5d7c70e079612ce19d508d79c78 100644 --- a/.gitignore +++ b/.gitignore @@ -193,5 +193,3 @@ TAGS # bazel symlinks bazel-* - -torch_npu/version.py \ No newline at end of file diff --git a/setup.py b/setup.py index 3f2f95419985cb6b9d775e13ed8ed87fb21174c6..431268ce49304d07df8a3e2fe234e2f30d83af6b 100644 --- a/setup.py +++ b/setup.py @@ -32,7 +32,7 @@ from setuptools import setup, find_packages, distutils, Extension BASE_DIR = os.path.dirname(os.path.abspath(__file__)) -VERSION = '1.8.1-rc1' +VERSION = '1.8.1rc1' def _get_build_mode(): diff --git a/torch_npu/__init__.py b/torch_npu/__init__.py index 8a9fa1cc128ddd0a39b285eb485834e4dc83f649..cedd54404fc7b8ddf019ccc4259a5594e2b8dc4b 100644 --- a/torch_npu/__init__.py +++ b/torch_npu/__init__.py @@ -24,6 +24,7 @@ import torch_npu.npu.amp import torch_npu.distributed import torch_npu._C +from .version import __version__ as __version__ __all__ = [] diff --git a/torch_npu/version.py b/torch_npu/version.py new file mode 100644 index 0000000000000000000000000000000000000000..32fb6cf7bf30cc67a5bf6bef7b2f2d33d6901856 --- /dev/null +++ b/torch_npu/version.py @@ -0,0 +1 @@ +__version__ = "1.8.1rc1"