[AMD][DSV4] Fix unified-KV pool sizing and SWA ring accounting (#30315)

2026-09-05 16:39:40 -07:00
parent 6a0c55fd6c
commit 514b45fd34
23 changed files with 792 additions and 160 deletions
@@ -131,6 +131,7 @@ class TestDeepSeekV4HiSparseAllocator(CustomTestCase):
        queue = DecodePreallocQueue.__new__(DecodePreallocQueue)
        queue.req_to_token_pool = req_to_token_pool
        queue.token_to_kv_pool_allocator = allocator
+        queue.token_to_kv_pool = None
        queue.tree_cache = SimpleNamespace(
            evictable_size=MagicMock(return_value=0),
            protected_size=MagicMock(return_value=0),