minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 13 Packages Projects Releases Wiki Activity

dsv4.1: Rust extension modules for image preprocessing, KV pool names, and PD bootstrap (#39677)

Browse Source
This commit is contained in:
Liangsheng Yin
2026-09-15 18:11:48 -07:00
committed by GitHub
parent dbd7281b06
commit 2f7d04da5d
5 changed files with 197 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rust/sglang-radix-tree/src/python_bindings.rs
+12
View File
@@ -241,6 +241,12 @@ fn pool_name_str(name: PoolName) -> &'static str {
PoolName::DeepseekV4C4Indexer => "deepseek_v4_c4_indexer",
PoolName::DeepseekV4C4IndexerScale => "deepseek_v4_c4_indexer_scale",
PoolName::DeepseekV4C128 => "deepseek_v4_c128",
PoolName::DeepseekV4C1 => "deepseek_v4_c1",
PoolName::DeepseekV4C1Indexer => "deepseek_v4_c1_indexer",
PoolName::DeepseekV4C1IndexerScale => "deepseek_v4_c1_indexer_scale",
PoolName::DeepseekV4C2 => "deepseek_v4_c2",
PoolName::DeepseekV4C2Indexer => "deepseek_v4_c2_indexer",
PoolName::DeepseekV4C2IndexerScale => "deepseek_v4_c2_indexer_scale",
PoolName::DeepseekV4C4State => "deepseek_v4_c4_state",
PoolName::DeepseekV4C4IndexerState => "deepseek_v4_c4_indexer_state",
PoolName::DeepseekV4C128State => "deepseek_v4_c128_state",
@@ -261,6 +267,12 @@ fn parse_pool_name(name: &str) -> PyResult<PoolName> {
"deepseek_v4_c4_indexer" => Ok(PoolName::DeepseekV4C4Indexer),
"deepseek_v4_c4_indexer_scale" => Ok(PoolName::DeepseekV4C4IndexerScale),
"deepseek_v4_c128" => Ok(PoolName::DeepseekV4C128),
"deepseek_v4_c1" => Ok(PoolName::DeepseekV4C1),
"deepseek_v4_c1_indexer" => Ok(PoolName::DeepseekV4C1Indexer),
"deepseek_v4_c1_indexer_scale" => Ok(PoolName::DeepseekV4C1IndexerScale),
"deepseek_v4_c2" => Ok(PoolName::DeepseekV4C2),
"deepseek_v4_c2_indexer" => Ok(PoolName::DeepseekV4C2Indexer),
"deepseek_v4_c2_indexer_scale" => Ok(PoolName::DeepseekV4C2IndexerScale),
"deepseek_v4_c4_state" => Ok(PoolName::DeepseekV4C4State),
"deepseek_v4_c4_indexer_state" => Ok(PoolName::DeepseekV4C4IndexerState),
"deepseek_v4_c128_state" => Ok(PoolName::DeepseekV4C128State),
rust/sglang-radix-tree/src/unified_tree_core.rs
+6
View File
@@ -342,6 +342,12 @@ pub enum PoolName {
DeepseekV4C4Indexer,
DeepseekV4C4IndexerScale,
DeepseekV4C128,
DeepseekV4C1,
DeepseekV4C1Indexer,
DeepseekV4C1IndexerScale,
DeepseekV4C2,
DeepseekV4C2Indexer,
DeepseekV4C2IndexerScale,
DeepseekV4C4State,
DeepseekV4C4IndexerState,
DeepseekV4C128State,