Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
4443b24b
Commit
4443b24b
authored
Jul 01, 2024
by
zhanggzh
Browse files
change get_dcu_version code
parent
281c1a93
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
2 deletions
+3
-2
nni/common/get_dcu_version.py
nni/common/get_dcu_version.py
+3
-2
No files found.
nni/common/get_dcu_version.py
View file @
4443b24b
...
...
@@ -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
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment