"vllm/vscode:/vscode.git/clone" did not exist on "6867bcd076191e8a60f0e49e90059b01910725be"
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)
if (major, minor) == ('2', '1'):
version = 'das.opt1.' + sha[:7] version = 'das.opt1.' + sha[:7]
if (major, minor) == ('2', '3'):
version = 'das.opt2.' + sha[:7]
else: else:
if (major, minor) == ('2', '1'):
version = 'das.opt1' 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"):
...@@ -339,6 +346,9 @@ def get_version_add(sha: Optional[str] = None) -> str: ...@@ -339,6 +346,9 @@ def get_version_add(sha: Optional[str] = None) -> str:
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")
file.write("__dcu_version__='0.5.0+{}'\n".format(version)) file.write("__dcu_version__='0.5.0+{}'\n".format(version))
......
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