move topk capturers to srt/state_capturer/ (#24450)

Co-authored-by: Yueming Yuan <yym022502@gmail.com>
Co-authored-by: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>
Co-authored-by: Ziang Li <ziangli@umich.edu>
This commit is contained in:
Liangsheng Yin
2026-05-05 15:54:01 -07:00
committed by GitHub
co-authored by Yueming Yuan fzyzcjy Ziang Li
parent 47a416fc62
commit 08d4c2072b
13 changed files with 28 additions and 28 deletions
@@ -5,7 +5,7 @@ import unittest
import aiohttp
import numpy as np
from sglang.srt.layers.attention.indexer_topk_capturer import (
from sglang.srt.state_capturer.indexer_topk import (
extract_indexer_topk_from_meta_info,
)
from sglang.srt.utils import kill_process_tree
@@ -9,7 +9,7 @@ import torch
from torch.nn.utils.rnn import pad_sequence
from sglang.benchmark.utils import download_and_cache_hf_file
from sglang.srt.layers.moe.routed_experts_capturer import (
from sglang.srt.state_capturer.routed_experts import (
extract_routed_experts_from_meta_info,
)
from sglang.srt.utils import kill_process_tree