20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[ROCm] Use device name in the warning (#15838)

Browse Source 
Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
This commit is contained in:
Gregory Shtrasberg 2025-04-01 03:10:48 -04:00 committed by GitHub
parent 4a9ce1784c
commit c7e63aa4d8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/engine/arg_utils.py
View File

@ -1530,7 +1530,7 @@ class EngineArgs:
# Non-CUDA is supported on V1, but off by default for now. # Non-CUDA is supported on V1, but off by default for now.
not_cuda = not current_platform.is_cuda() not_cuda = not current_platform.is_cuda()
if not_cuda and _warn_or_fallback( # noqa: SIM103 if not_cuda and _warn_or_fallback( # noqa: SIM103
current_platform.device_type): current_platform.device_name):
return False return False
############################################################# #############################################################