Skip to content

GitLab

Commit a4456dfa authored by zhuwenwen's avatar zhuwenwen
Browse files

update version

parent dfdc05ae
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 4 deletions
setup.py
View file @ a4456dfa
...@@ -336,12 +336,9 @@ def get_version_add(sha: Optional[str] = None) -> str: ...@@ -336,12 +336,9 @@ def get_version_add(sha: Optional[str] = None) -> str:
rocm_version_path = os.path.join(rocm_path, '.info', "rocm_version") rocm_version_path = os.path.join(rocm_path, '.info', "rocm_version")
with open(rocm_version_path, 'r',encoding='utf-8') as file: with open(rocm_version_path, 'r',encoding='utf-8') as file:
lines = file.readlines() lines = file.readlines()
rocm_version=lines[0][:-2].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