mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
e69c2f85f6
Summary: Add a `version_info` similar to `sys.version_info` for being able to make version tests. Example generated `version.py`: ``` __version__ = '1.8.0a0' version_info = (1, 8, 0, 'a0') # or version_info = (1, 8, 0, 'a0', 'deadbeef') if you're in a Git checkout debug = False cuda = None git_version = '671ee71ad4b6f507218d1cad278a8e743780b716' hip = None ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/48414 Reviewed By: zhangguanheng66 Differential Revision: D25416620 Pulled By: malfet fbshipit-source-id: 20b561a0c76ac0b16ff92f4bd43f8b724971e444
64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
import argparse
|
|
import os
|
|
import subprocess
|
|
from pathlib import Path
|
|
from distutils.util import strtobool
|
|
|
|
def get_sha():
|
|
try:
|
|
return subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=pytorch_root).decode('ascii').strip()
|
|
except Exception:
|
|
return 'Unknown'
|
|
|
|
def get_torch_version(sha=None):
|
|
pytorch_root = Path(__file__).parent.parent
|
|
version = open('version.txt', 'r').read().strip()
|
|
if sha is None:
|
|
sha = get_sha()
|
|
|
|
if os.getenv('PYTORCH_BUILD_VERSION'):
|
|
assert os.getenv('PYTORCH_BUILD_NUMBER') is not None
|
|
build_number = int(os.getenv('PYTORCH_BUILD_NUMBER'))
|
|
version = os.getenv('PYTORCH_BUILD_VERSION')
|
|
if build_number > 1:
|
|
version += '.post' + str(build_number)
|
|
elif sha != 'Unknown':
|
|
version += '+' + sha[:7]
|
|
|
|
first_non_numeric = min(i for i, c in enumerate(version) if c not in "0123456789.")
|
|
version_suffix = version[first_non_numeric:]
|
|
version_info = tuple(
|
|
[int(part) for part in version[:first_non_numeric].split(".")]
|
|
+ version_suffix.split("+")
|
|
)
|
|
|
|
return version, version_info
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Generate torch/version.py from build and environment metadata.")
|
|
parser.add_argument("--is_debug", type=strtobool, help="Whether this build is debug mode or not.")
|
|
parser.add_argument("--cuda_version", type=str)
|
|
parser.add_argument("--hip_version", type=str)
|
|
|
|
args = parser.parse_args()
|
|
|
|
assert args.is_debug is not None
|
|
args.cuda_version = None if args.cuda_version == '' else args.cuda_version
|
|
args.hip_version = None if args.hip_version == '' else args.hip_version
|
|
|
|
pytorch_root = Path(__file__).parent.parent
|
|
version_path = pytorch_root / "torch" / "version.py"
|
|
sha = get_sha()
|
|
version, version_info = get_torch_version(sha)
|
|
|
|
with open(version_path, 'w') as f:
|
|
f.write("__version__ = '{}'\n".format(version))
|
|
f.write("version_info = {}\n".format(version_info))
|
|
# NB: This is not 100% accurate, because you could have built the
|
|
# library code with DEBUG, but csrc without DEBUG (in which case
|
|
# this would claim to be a release build when it's not.)
|
|
f.write("debug = {}\n".format(repr(bool(args.is_debug))))
|
|
f.write("cuda = {}\n".format(repr(args.cuda_version)))
|
|
f.write("git_version = {}\n".format(repr(sha)))
|
|
f.write("hip = {}\n".format(repr(args.hip_version)))
|