Commit b240f635 authored by zhuwenwen's avatar zhuwenwen
Browse files

add torch version

parent 4ac087d9
...@@ -322,13 +322,20 @@ def get_sha(root: Union[str, Path]) -> str: ...@@ -322,13 +322,20 @@ def get_sha(root: Union[str, Path]) -> str:
def get_version_add(sha: Optional[str] = None) -> str: def get_version_add(sha: Optional[str] = None) -> str:
vllm_root = os.path.dirname(os.path.abspath(__file__)) vllm_root = os.path.dirname(os.path.abspath(__file__))
add_version_path = os.path.join(os.path.join(vllm_root, "vllm"), "version.py") add_version_path = os.path.join(os.path.join(vllm_root, "vllm"), "version.py")
major, minor, *res = torch.__version__.split('.')
if add_git_version: if add_git_version:
if sha != 'Unknown': if sha != 'Unknown':
if sha is None: if sha is None:
sha = get_sha(vllm_root) sha = get_sha(vllm_root)
version = 'das.opt1.' + sha[:7] if (major, minor) == ('2', '1'):
version = 'das.opt1.' + sha[:7]
if (major, minor) == ('2', '3'):
version = 'das.opt2.' + sha[:7]
else: else:
version = 'das.opt1' if (major, minor) == ('2', '1'):
version = 'das.opt1'
if (major, minor) == ('2', '3'):
version = 'das.opt2'
# dtk version # dtk version
if os.getenv("ROCM_PATH"): if os.getenv("ROCM_PATH"):
...@@ -338,6 +345,9 @@ def get_version_add(sha: Optional[str] = None) -> str: ...@@ -338,6 +345,9 @@ def get_version_add(sha: Optional[str] = None) -> str:
lines = file.readlines() lines = file.readlines()
rocm_version=lines[0].replace(".", "") rocm_version=lines[0].replace(".", "")
version += ".dtk" + rocm_version version += ".dtk" + rocm_version
# torch version
version += ".torch" + torch.__version__[:5]
with open(add_version_path, encoding="utf-8",mode="w") as file: with open(add_version_path, encoding="utf-8",mode="w") as file:
file.write("__version__='0.5.0'\n") file.write("__version__='0.5.0'\n")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment