chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

Fix NSA indexer test and move it to pre commit test (#17682)

Browse Source
This commit is contained in:
Kangyan-Zhou
2026-01-24 12:06:18 -08:00
committed by GitHub
parent 8656a146a6
commit 137eb5b95c
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/registered/kernels/test_nsa_indexer.py
View File

@@ -24,7 +24,7 @@ from sglang.srt.model_executor.forward_batch_info import ForwardBatch, ForwardMo
from sglang.srt.server_args import ServerArgs, set_global_server_args_for_scheduler
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=2, suite="nightly-1-gpu", nightly=True)
register_cuda_ci(est_time=2, suite="stage-b-test-small-1-gpu")
# Global configuration for all indexer tests
DEFAULT_CONFIG = {
@@ -41,6 +41,7 @@ DEFAULT_CONFIG = {
"q_lora_rank": 1536,
"kv_lora_rank": 512,
"qk_rope_head_dim": 64,
"qk_nope_head_dim": 128,
"max_position_embeddings": 163840,
"rope_theta": 10000.0,
"layer_id": 0,
@@ -190,6 +191,7 @@ class MockModelRunner:
"num_attention_heads": 128,
"kv_lora_rank": self.config["kv_lora_rank"],
"qk_rope_head_dim": self.config["qk_rope_head_dim"],
"qk_nope_head_dim": self.config["qk_nope_head_dim"],
"hf_config": hf_config,
},
)()