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
b240f635
"vllm/vscode:/vscode.git/clone" did not exist on "6867bcd076191e8a60f0e49e90059b01910725be"
Commit
b240f635
authored
Aug 29, 2024
by
zhuwenwen
Browse files
add torch version
parent
4ac087d9
Changes
1
Show 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 @
b240f635
...
@@ -322,13 +322,20 @@ def get_sha(root: Union[str, Path]) -> str:
...
@@ -322,13 +322,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
)
if
(
major
,
minor
)
==
(
'2'
,
'1'
):
version
=
'das.opt1.'
+
sha
[:
7
]
version
=
'das.opt1.'
+
sha
[:
7
]
if
(
major
,
minor
)
==
(
'2'
,
'3'
):
version
=
'das.opt2.'
+
sha
[:
7
]
else
:
else
:
if
(
major
,
minor
)
==
(
'2'
,
'1'
):
version
=
'das.opt1'
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"
):
...
@@ -339,6 +346,9 @@ def get_version_add(sha: Optional[str] = None) -> str:
...
@@ -339,6 +346,9 @@ def get_version_add(sha: Optional[str] = None) -> str:
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
]
with
open
(
add_version_path
,
encoding
=
"utf-8"
,
mode
=
"w"
)
as
file
:
with
open
(
add_version_path
,
encoding
=
"utf-8"
,
mode
=
"w"
)
as
file
:
file
.
write
(
"__version__='0.5.0'
\n
"
)
file
.
write
(
"__version__='0.5.0'
\n
"
)
file
.
write
(
"__dcu_version__='0.5.0+{}'
\n
"
.
format
(
version
))
file
.
write
(
"__dcu_version__='0.5.0+{}'
\n
"
.
format
(
version
))
...
...
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