From a0563e7368e25663345780b31bb23a60d828ca09 Mon Sep 17 00:00:00 2001
From: yuantao <2422264527@qq.com>
Date: Sat, 13 Dec 2025 16:32:56 +0800
Subject: [PATCH] Add assert in init of SinkFullAttentionManager

Signed-off-by: yuantao <2422264527@qq.com>
---
 vllm/v1/core/single_type_kv_cache_manager.py | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/vllm/v1/core/single_type_kv_cache_manager.py b/vllm/v1/core/single_type_kv_cache_manager.py
index 4eeea533464c9..14905b36754b4 100644
--- a/vllm/v1/core/single_type_kv_cache_manager.py
+++ b/vllm/v1/core/single_type_kv_cache_manager.py
@@ -797,12 +797,9 @@ class SinkFullAttentionManager(FullAttentionManager):
             kv_cache_spec, block_pool, kv_cache_group_id, dcp_world_size, pcp_world_size
         )
         sink_len = kv_cache_spec.sink_len
-        if sink_len > 0:
-            assert sink_len % self.block_size == 0
-            num_sink_block = sink_len // self.block_size
-            self.sink_blocks = self.block_pool.free_block_queue.popleft_n(
-                num_sink_block
-            )
+        assert sink_len is not None and sink_len > 0 and sink_len % self.block_size == 0
+        num_sink_block = sink_len // self.block_size
+        self.sink_blocks = self.block_pool.free_block_queue.popleft_n(num_sink_block)
 
 
 spec_manager_map: dict[type[KVCacheSpec], type[SingleTypeKVCacheManager]] = {