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
dafb4e50
Unverified
Commit
dafb4e50
authored
Mar 19, 2025
by
Roger Wang
Committed by
GitHub
Mar 19, 2025
Browse files
[V1][Bugfix] Fix oracle for device checking (#15104)
Signed-off-by:
Roger Wang
<
ywang@roblox.com
>
parent
68cf1601
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
0 deletions
+7
-0
vllm/engine/arg_utils.py
vllm/engine/arg_utils.py
+7
-0
No files found.
vllm/engine/arg_utils.py
View file @
dafb4e50
...
@@ -1578,6 +1578,13 @@ class EngineArgs:
...
@@ -1578,6 +1578,13 @@ class EngineArgs:
_raise_or_fallback
(
feature_name
=
name
,
recommend_to_remove
=
True
)
_raise_or_fallback
(
feature_name
=
name
,
recommend_to_remove
=
True
)
return
False
return
False
# No support for device type other than CUDA, AMD (experiemntal) or
# TPU (experimental) so far.
if
not
(
current_platform
.
is_cuda_alike
()
or
current_platform
.
is_tpu
()):
_raise_or_fallback
(
feature_name
=
f
"device type=
{
current_platform
.
device_type
}
"
,
recommend_to_remove
=
False
)
return
False
#############################################################
#############################################################
# Experimental Features - allow users to opt in.
# Experimental Features - allow users to opt in.
...
...
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