Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
55588cbd
"vscode:/vscode.git/clone" did not exist on "6f2909405edbd39478891f71dc2bfcbcda8530fa"
Commit
55588cbd
authored
Aug 29, 2024
by
zhuwenwen
Browse files
add torch version
parent
35068264
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
12 additions
and
2 deletions
+12
-2
setup.py
setup.py
+12
-2
No files found.
setup.py
View file @
55588cbd
...
@@ -364,13 +364,20 @@ def get_sha(root: Union[str, Path]) -> str:
...
@@ -364,13 +364,20 @@ def get_sha(root: Union[str, Path]) -> str:
def
get_version_add
(
sha
:
Optional
[
str
]
=
None
)
->
str
:
def
get_version_add
(
sha
:
Optional
[
str
]
=
None
)
->
str
:
vllm_root
=
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
))
vllm_root
=
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
))
add_version_path
=
os
.
path
.
join
(
os
.
path
.
join
(
vllm_root
,
"vllm"
),
"version.py"
)
add_version_path
=
os
.
path
.
join
(
os
.
path
.
join
(
vllm_root
,
"vllm"
),
"version.py"
)
major
,
minor
,
*
res
=
torch
.
__version__
.
split
(
'.'
)
if
add_git_version
:
if
add_git_version
:
if
sha
!=
'Unknown'
:
if
sha
!=
'Unknown'
:
if
sha
is
None
:
if
sha
is
None
:
sha
=
get_sha
(
vllm_root
)
sha
=
get_sha
(
vllm_root
)
version
=
'das.opt1.'
+
sha
[:
7
]
if
(
major
,
minor
)
==
(
'2'
,
'1'
):
version
=
'das.opt1.'
+
sha
[:
7
]
if
(
major
,
minor
)
==
(
'2'
,
'3'
):
version
=
'das.opt2.'
+
sha
[:
7
]
else
:
else
:
version
=
'das.opt1'
if
(
major
,
minor
)
==
(
'2'
,
'1'
):
version
=
'das.opt1'
if
(
major
,
minor
)
==
(
'2'
,
'3'
):
version
=
'das.opt2'
# dtk version
# dtk version
if
os
.
getenv
(
"ROCM_PATH"
):
if
os
.
getenv
(
"ROCM_PATH"
):
...
@@ -380,6 +387,9 @@ def get_version_add(sha: Optional[str] = None) -> str:
...
@@ -380,6 +387,9 @@ def get_version_add(sha: Optional[str] = None) -> str:
lines
=
file
.
readlines
()
lines
=
file
.
readlines
()
rocm_version
=
lines
[
0
].
replace
(
"."
,
""
)
rocm_version
=
lines
[
0
].
replace
(
"."
,
""
)
version
+=
".dtk"
+
rocm_version
version
+=
".dtk"
+
rocm_version
# torch version
version
+=
".torch"
+
torch
.
__version__
[:
5
]
new_version_content
=
f
"""
new_version_content
=
f
"""
import warnings
import warnings
...
...
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