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
50fede66
Unverified
Commit
50fede66
authored
Aug 26, 2025
by
Cyrus Leung
Committed by
GitHub
Aug 26, 2025
Browse files
[V1] Enable V1 for compute capability < 8.0 + FP32 (#23614)
Signed-off-by:
DarkLight1337
<
tlleungac@connect.ust.hk
>
parent
b5d34af3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
8 deletions
+8
-8
vllm/engine/arg_utils.py
vllm/engine/arg_utils.py
+8
-8
No files found.
vllm/engine/arg_utils.py
View file @
50fede66
...
...
@@ -1433,15 +1433,15 @@ class EngineArgs:
recommend_to_remove
=
True
)
return
False
# Need at least Ampere for now (FA support required).
# Skip this check if we are running on a non-GPU platform,
# or if the device capability is not available
# (e.g. in a Ray actor without GPUs).
# Triton v3.3 has f16 conversion regression issue on Turing and Volta,
# which broke fp16 inference
# see: https://github.com/triton-lang/triton/issues/6698
if
(
current_platform
.
is_cuda
()
and
current_platform
.
get_device_capability
()
and
current_platform
.
get_device_capability
().
major
<
8
):
_raise_or_fallback
(
feature_name
=
"Compute Capability < 8.0"
,
recommend_to_remove
=
False
)
and
not
current_platform
.
has_device_capability
(
80
)
and
model_config
.
dtype
==
torch
.
float16
):
_raise_or_fallback
(
feature_name
=
"Compute Capability < 8.0 with FP16"
,
recommend_to_remove
=
False
)
return
False
if
self
.
kv_cache_dtype
!=
"auto"
:
...
...
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