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
660af62e
Commit
660af62e
authored
May 07, 2025
by
zhuwenwen
Browse files
update rocm.py
parent
c7096363
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
vllm/platforms/rocm.py
vllm/platforms/rocm.py
+3
-3
No files found.
vllm/platforms/rocm.py
View file @
660af62e
...
...
@@ -201,9 +201,9 @@ class RocmPlatform(Platform):
# f"does not support block size {block_size}."
# "(currently only supports block size 1)")
# else:
raise
ValueError
(
f
" The selected backend,
{
selected_backend
.
name
}
,"
f
"is not MLA type while requested for MLA backend."
)
#
raise ValueError(
#
f" The selected backend, {selected_backend.name},"
#
f"is not MLA type while requested for MLA backend.")
selected_backend
=
(
_Backend
.
ROCM_FLASH
if
selected_backend
==
_Backend
.
FLASH_ATTN
else
selected_backend
)
...
...
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