Commit 4443b24b authored by zhanggzh's avatar zhanggzh
Browse files

change get_dcu_version code

parent 281c1a93
......@@ -25,7 +25,8 @@ def abi_value():
def dtk_version_value():
try:
dtk_path=os.getenv('ROCM_PATH')
#dtk_path=os.getenv('ROCM_PATH')
dtk_path='/opt/dtk-24.04'
dtk_version_path = os.path.join(dtk_path, '.info', "version-dev")
with open(dtk_version_path, 'r',encoding='utf-8') as file:
lines = file.readlines()
......@@ -49,7 +50,7 @@ def dcu_version():
try:
release = os.environ.get('NNI_RELEASE')
nni_version = release or UNKNOWN
dcu_version = nni_version + "das1.1" + nni_whl_name()
dcu_version = nni_version + "+" "das1.1" + nni_whl_name()
return dcu_version
except Exception:
return UNKNOWN
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