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
973c9d01
Unverified
Commit
973c9d01
authored
Sep 13, 2025
by
Victor Ziliang Peng
Committed by
GitHub
Sep 13, 2025
Browse files
[Minor] Simplify duplicative device check for cuda (#24793)
Signed-off-by:
Ziliang Peng
<
ziliangdotme@gmail.com
>
parent
15b8fef4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
vllm/platforms/cuda.py
vllm/platforms/cuda.py
+1
-2
No files found.
vllm/platforms/cuda.py
View file @
973c9d01
...
...
@@ -64,8 +64,7 @@ class CudaPlatformBase(Platform):
if
self
.
has_device_capability
(
80
):
# Ampere and Hopper or later NVIDIA GPUs.
return
[
torch
.
bfloat16
,
torch
.
float16
,
torch
.
float32
]
elif
(
not
self
.
has_device_capability
(
80
)
)
and
self
.
has_device_capability
(
60
):
if
self
.
has_device_capability
(
60
):
# Pascal, Volta and Turing NVIDIA GPUs, BF16 is not supported
return
[
torch
.
float16
,
torch
.
float32
]
# Kepler and Maxwell NVIDIA GPUs, only FP32 is supported,
...
...
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