Skip to content

GitLab

Commit c11ff52c authored by zhanggezhong's avatar zhanggezhong
Browse files

Update get_dcu_version.py

parent 472027a2
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
get_dcu_version.py
View file @ c11ff52c
...@@ -7,10 +7,10 @@ from pathlib import Path ...@@ -7,10 +7,10 @@ from pathlib import Path
import os import os
UNKNOWN = "Unknown" UNKNOWN = "Unknown"
def sha_value(tf_root): def sha_value(moe_root):
try: try:
return ( return (
subprocess.check_output(["git", "rev-parse", "HEAD"], cwd=tf_root) subprocess.check_output(["git", "rev-parse", "HEAD"], cwd=moe_root)
.decode("ascii") .decode("ascii")
.strip() .strip()
) )
...@@ -48,8 +48,8 @@ def torch_version_value(): ...@@ -48,8 +48,8 @@ def torch_version_value():
def moe_whl_name(): def moe_whl_name():
try: try:
tf_root = Path(__file__).parent.parent moe_root = Path(__file__).parent
sha = "das1.1.git" + sha_value(tf_root)[0:7] sha = "das1.1.git" + sha_value(moe_root)[0:7]
abi = "abi" + abi_value() abi = "abi" + abi_value()
dtk_version = dtk_version_value() dtk_version = dtk_version_value()
try: try:
......
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