Skip to content

GitLab

Unverified Commit 43c57925 authored by Micah Williamson's avatar Micah Williamson Committed by GitHub
Browse files

[ROCm][CI] Fix test_cpu_offloading for ROCm (#29548) 


Signed-off-by: default avatarMicah Williamson <micah.williamson@amd.com>
parent 3ecabd06
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/v1/kv_offload/test_cpu_offloading.py
View file @ 43c57925
......@@ -20,6 +20,8 @@ ATTN_BACKENDS = ["FLASH_ATTN"]
if current_platform.is_cuda():
ATTN_BACKENDS.append("FLASHINFER")
elif current_platform.is_rocm():
ATTN_BACKENDS = ["TRITON_ATTN"]
class MockSubscriber:
......
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