Skip to content

Commit c7e63aa

Browse files
authored
[ROCm] Use device name in the warning (#15838)
Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
1 parent 4a9ce17 commit c7e63aa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/engine/arg_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1530,7 +1530,7 @@ def _is_v1_supported_oracle(self, model_config: ModelConfig) -> bool:
15301530
# Non-CUDA is supported on V1, but off by default for now.
15311531
not_cuda = not current_platform.is_cuda()
15321532
if not_cuda and _warn_or_fallback( # noqa: SIM103
1533-
current_platform.device_type):
1533+
current_platform.device_name):
15341534
return False
15351535
#############################################################
15361536

0 commit comments

Comments
 (0)