diff --git a/python/sglang/srt/layers/attention/nsa_backend.py b/python/sglang/srt/layers/attention/nsa_backend.py index 06a1259a2..e160cc400 100644 --- a/python/sglang/srt/layers/attention/nsa_backend.py +++ b/python/sglang/srt/layers/attention/nsa_backend.py @@ -37,7 +37,6 @@ from sglang.srt.layers.attention.nsa.cp_shared_kv_runtime import ( mask_batch_logical_pages_to_valid_lengths, materialize_prefix_and_reuse_current_kv_page_slots, materialize_shared_token_kv_buffer, - maybe_build_current_page_writer_ranks, pack_current_mla_kv_for_reuse, tensor_debug_checksum, tensor_debug_summary, @@ -2115,19 +2114,6 @@ class NativeSparseAttnBackend( current_slot_spans=current_slot_spans, layer_id=layer.layer_id, nvtx_source="mla.current_only_page_slots", - current_page_writer_ranks=( - maybe_build_current_page_writer_ranks( - forward_batch=forward_batch, - prefix_lens_cpu=getattr( - forward_batch, "extend_prefix_lens_cpu", None - ), - extend_lens_cpu=getattr( - forward_batch, "extend_seq_lens_cpu", None - ), - page_size=page_size, - layout=forward_batch.cp_shared_kv_layout, - ) - ), ) ) else: @@ -2274,15 +2260,6 @@ class NativeSparseAttnBackend( prefix_slot_spans=prefix_slot_spans, current_slot_spans=current_slot_spans, layer_id=layer.layer_id, - current_page_writer_ranks=( - maybe_build_current_page_writer_ranks( - forward_batch=forward_batch, - prefix_lens_cpu=prefix_lens_cpu, - extend_lens_cpu=extend_lens_cpu, - page_size=page_size, - layout=forward_batch.cp_shared_kv_layout, - ) - ), ) ) if (