Unverified Commit 5241ca50 authored by Gregory Shtrasberg's avatar Gregory Shtrasberg Committed by GitHub
Browse files

[ROCm][V1] Adding ROCm to the list of plaforms using V1 by default (#19440)


Signed-off-by: default avatarGregory Shtrasberg <Gregory.Shtrasberg@amd.com>
parent da9b523c
...@@ -1442,7 +1442,7 @@ class EngineArgs: ...@@ -1442,7 +1442,7 @@ class EngineArgs:
# Non-[CUDA, TPU] may be supported on V1, but off by default for now. # Non-[CUDA, TPU] may be supported on V1, but off by default for now.
v0_hardware = not any( v0_hardware = not any(
(current_platform.is_cuda(), current_platform.is_tpu(), (current_platform.is_cuda_alike(), current_platform.is_tpu(),
(current_platform.is_cpu() (current_platform.is_cpu()
and current_platform.get_cpu_architecture() == CpuArchEnum.X86))) and current_platform.get_cpu_architecture() == CpuArchEnum.X86)))
if v0_hardware and _warn_or_fallback( # noqa: SIM103 if v0_hardware and _warn_or_fallback( # noqa: SIM103
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment