Skip to content

Commit c8aa1a7

Browse files
zhangxinyuanliuhengyuwhx-sjtucursoragent
authored
[feat][sgl-atom] add qwen3 sglang dense model support (#1416)
* fix Qwen3 dense startup with ATOM backend Register Qwen3 dense for the SGLang ATOM wrapper and route page-size 1 dense decode through the existing native AITER path to avoid invalid KV layout reshapes during CUDA graph capture. Co-authored-by: Cursor <cursoragent@cursor.com> * Limit Qwen3 dense startup fix scope Restore the existing dense attention routing and use page-size 16 only for Qwen3-32B MI308 SGLang benchmark entries so the model stays on the ATOM attention path. Co-authored-by: Cursor <cursoragent@cursor.com> --------- Co-authored-by: Hexiang Wang <56632993+whx-sjtu@users.noreply.github.com> Co-authored-by: Cursor <cursoragent@cursor.com>
1 parent e2542e4 commit c8aa1a7

2 files changed

Lines changed: 4 additions & 0 deletions

File tree

.github/benchmark/sglang_benchmark_models.json

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -955,6 +955,7 @@
955955
"prefix": "qwen3-32b-fp8-tp1-mi308",
956956
"extra_args": [
957957
"--tensor-parallel-size 1",
958+
"--page-size 16",
958959
"qwen_reasoning"
959960
],
960961
"bench_args": "",
@@ -971,6 +972,7 @@
971972
"prefix": "qwen3-32b-fp8-tp8-mi308",
972973
"extra_args": [
973974
"--tensor-parallel-size 8",
975+
"--page-size 16",
974976
"qwen_reasoning"
975977
],
976978
"bench_args": "",

atom/plugin/sglang/runtime/model_arch.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,7 @@ def _install_deepseek_v4_adapters(model: Any) -> None:
8686
prepare_config=_prepare_kimi_k25_config,
8787
install_adapters=_install_deepseek_mla_adapters,
8888
),
89+
"Qwen3ForCausalLM": SGLangModelAdapterSpec(),
8990
"Qwen3MoeForCausalLM": SGLangModelAdapterSpec(),
9091
"Qwen3NextForCausalLM": SGLangModelAdapterSpec(
9192
wrapper_binds_gdn_context=True,
@@ -114,6 +115,7 @@ def _install_deepseek_v4_adapters(model: Any) -> None:
114115
"DeepseekV3ForCausalLM",
115116
"DeepseekV32ForCausalLM",
116117
"GlmMoeDsaForCausalLM",
118+
"Qwen3ForCausalLM",
117119
"Qwen3MoeForCausalLM",
118120
"Qwen3NextForCausalLM",
119121
"MiniMaxM2ForCausalLM",

0 commit comments

Comments
 (0)