From 087a138963192749043e306a0b2552c6ba465d8e Mon Sep 17 00:00:00 2001 From: Andreas Karatzas Date: Wed, 7 Jan 2026 22:35:25 -0600 Subject: [PATCH] [ROCm][CI] Fix attention backend test flakiness from uninitialized KV cache memory (#31928) Signed-off-by: Andreas Karatzas --- tests/v1/attention/test_attention_backends.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/v1/attention/test_attention_backends.py b/tests/v1/attention/test_attention_backends.py index 2b9b0dc1e..3a7e0aa66 100644 --- a/tests/v1/attention/test_attention_backends.py +++ b/tests/v1/attention/test_attention_backends.py @@ -136,7 +136,7 @@ def create_and_prepopulate_kv_cache( slot_mapping = common_attn_metadata.slot_mapping # Create KV cache - kv_cache = torch.empty( + kv_cache = torch.zeros( 2, num_blocks, block_size, num_kv_heads, head_size, dtype=dtype, device=device ) kv_cache_flat = kv_cache.view(2, -1, num_kv_heads, head_size)