Commit f4bc5080 authored by gaoqiong's avatar gaoqiong
Browse files

Update setup.py

parent e5d5fc7e
......@@ -77,7 +77,7 @@ def get_version_add(sha: Optional[str] = None) -> str:
rocm_version_path = os.path.join(rocm_path, '.info', "rocm_version")
with open(rocm_version_path, 'r',encoding='utf-8') as file:
lines = file.readlines()
rocm_version=lines[0][:-2].replace(".", "")
rocm_version=lines[0][:].replace(".", "")
version += ".dtk" + rocm_version
# torch 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