Commit 163b243a authored by zhuwenwen's avatar zhuwenwen
Browse files

remove torch version

parent 6bb59e94
...@@ -375,20 +375,12 @@ def get_sha(root: Union[str, Path]) -> str: ...@@ -375,20 +375,12 @@ 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]
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"):
...@@ -398,9 +390,6 @@ def get_version_add(sha: Optional[str] = None) -> str: ...@@ -398,9 +390,6 @@ 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]
new_version_content = f""" new_version_content = f"""
import warnings import warnings
......
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