diff --git a/python/sglang/srt/layers/attention/nsa/cp_shared_kv_prefetch.py b/python/sglang/srt/layers/attention/nsa/cp_shared_kv_prefetch.py index 3a8cf61f9..69c00c05f 100644 --- a/python/sglang/srt/layers/attention/nsa/cp_shared_kv_prefetch.py +++ b/python/sglang/srt/layers/attention/nsa/cp_shared_kv_prefetch.py @@ -178,9 +178,6 @@ class CpSharedKVMlaPrefetcher: if token_to_kv_pool is None: _prefetch_log("create_skip reason=missing_token_to_kv_pool") return None - if getattr(token_to_kv_pool, "layer_transfer_counter", None) is not None: - _prefetch_log("create_skip reason=layer_transfer_active") - return None layout = getattr(forward_batch, "cp_shared_kv_layout", None) if layout is None: @@ -705,12 +702,6 @@ class CpSharedKVIndexPrefetcher: "forward batch has no token_to_kv_pool.", ) return None - if getattr(token_to_kv_pool, "layer_transfer_counter", None) is not None: - _index_prefetch_fallback_log( - "layer_transfer_active", - "layer transfer is active.", - ) - return None layout = getattr(forward_batch, "cp_shared_kv_layout", None) if layout is None: