diff --git a/python/sglang/srt/managers/cache_controller.py b/python/sglang/srt/managers/cache_controller.py index bd2f170b0..89d4c088e 100644 --- a/python/sglang/srt/managers/cache_controller.py +++ b/python/sglang/srt/managers/cache_controller.py @@ -1412,8 +1412,6 @@ class HiCacheController: page_size=page_size, object_ranges=object_ranges, required_payloads=tuple(required_payloads), - committed_payload_layers={payload: set() for payload in required_payloads}, - committed=False, object_key=object_key, ) return HiCacheWriteReservation( diff --git a/python/sglang/srt/mem_cache/cp_shared_l2_pool.py b/python/sglang/srt/mem_cache/cp_shared_l2_pool.py index 15d0de90b..1a04ae211 100644 --- a/python/sglang/srt/mem_cache/cp_shared_l2_pool.py +++ b/python/sglang/srt/mem_cache/cp_shared_l2_pool.py @@ -438,8 +438,6 @@ class CpSharedL2NodeMetadata: page_size: int object_ranges: dict[str, CpSharedL2ObjectRange] required_payloads: tuple[str, ...] - committed_payload_layers: dict[str, set[int]] - committed: bool = False object_key: str = "" def __post_init__(self): @@ -466,13 +464,6 @@ class CpSharedL2NodeMetadata: f"payload_kind={object_range.payload_kind!r}" ) - for payload_kind, layers in self.committed_payload_layers.items(): - self._validate_payload_kind(payload_kind) - if not isinstance(layers, set): - raise ValueError("committed_payload_layers values must be sets") - if any(layer < 0 for layer in layers): - raise ValueError("committed payload layer ids must be non-negative") - @staticmethod def _validate_payload_kind(payload_kind: str): if payload_kind not in _VALID_PAYLOAD_KINDS: @@ -533,10 +524,6 @@ class CpSharedL2NodeMetadata: generation=object_range.generation, ) - committed_layers = { - payload_kind: set(layers) - for payload_kind, layers in self.committed_payload_layers.items() - } return ( CpSharedL2NodeMetadata( logical_len=split_len, @@ -544,11 +531,6 @@ class CpSharedL2NodeMetadata: page_size=self.page_size, object_ranges=parent_ranges, required_payloads=tuple(self.required_payloads), - committed_payload_layers={ - payload_kind: set(layers) - for payload_kind, layers in committed_layers.items() - }, - committed=self.committed, object_key=parent_key, ), CpSharedL2NodeMetadata( @@ -557,11 +539,6 @@ class CpSharedL2NodeMetadata: page_size=self.page_size, object_ranges=child_ranges, required_payloads=tuple(self.required_payloads), - committed_payload_layers={ - payload_kind: set(layers) - for payload_kind, layers in committed_layers.items() - }, - committed=self.committed, object_key=child_key, ), ) diff --git a/test/registered/unit/mem_cache/test_cp_shared_l2_pool.py b/test/registered/unit/mem_cache/test_cp_shared_l2_pool.py index f00ef3b5e..20345c08b 100644 --- a/test/registered/unit/mem_cache/test_cp_shared_l2_pool.py +++ b/test/registered/unit/mem_cache/test_cp_shared_l2_pool.py @@ -512,10 +512,8 @@ class TestCpSharedL2Metadata(unittest.TestCase): page_size=64, object_ranges=ranges, required_payloads=(PAYLOAD_TARGET_KV, PAYLOAD_INDEX_K), - committed_payload_layers={PAYLOAD_TARGET_KV: {0, 1}, PAYLOAD_INDEX_K: {0}}, ) - self.assertFalse(metadata.committed) self.assertIs( metadata.object_ranges[PAYLOAD_TARGET_KV], ranges[PAYLOAD_TARGET_KV] ) @@ -554,7 +552,6 @@ class TestCpSharedL2Metadata(unittest.TestCase): ) }, required_payloads=(PAYLOAD_TARGET_KV,), - committed_payload_layers={}, ) def test_shared_metadata_split_partitions_payload_ranges_by_page(self): @@ -581,11 +578,6 @@ class TestCpSharedL2Metadata(unittest.TestCase): ), }, required_payloads=(PAYLOAD_TARGET_KV, PAYLOAD_DRAFT_KV), - committed_payload_layers={ - PAYLOAD_TARGET_KV: {0, 1}, - PAYLOAD_DRAFT_KV: {0}, - }, - committed=True, object_key="node-10", ) @@ -595,8 +587,6 @@ class TestCpSharedL2Metadata(unittest.TestCase): self.assertEqual(parent.padded_len, 128) self.assertEqual(child.logical_len, 64) self.assertEqual(child.padded_len, 128) - self.assertTrue(parent.committed) - self.assertTrue(child.committed) self.assertEqual(parent.required_payloads, metadata.required_payloads) self.assertEqual(child.required_payloads, metadata.required_payloads) self.assertEqual( @@ -660,7 +650,6 @@ class TestCpSharedL2Metadata(unittest.TestCase): ) }, required_payloads=(PAYLOAD_TARGET_KV,), - committed_payload_layers={PAYLOAD_TARGET_KV: {0}}, object_key="node-11", )