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

Signed-off-by: Micah Williamson <micah.williamson@amd.com>
2025-11-27 01:54:44 -06:00
parent 3ecabd06ee
commit 43c5792592
1 changed files with 2 additions and 0 deletions
--- a/tests/v1/kv_offload/test_cpu_offloading.py
+++ b/tests/v1/kv_offload/test_cpu_offloading.py
@@ -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: