chore: move cuda_vmm_utils.py under srt/utils/ - #36053
Merged
Merged
Conversation
Relocate python/sglang/srt/cuda_vmm_utils.py to python/sglang/srt/utils/cuda_vmm_utils.py, next to the related cuda_vmm_transport_utils.py / cuda_ipc_transport_utils.py, and update all import sites. Pure move: the file contents are unchanged.
merrymercy
requested review from
BBuf,
Edwardf0t1,
Fridge003,
HaiShaw,
JustinTong0323,
Ying1123,
alphabetc1,
ch-wan,
hanming-lu,
hnyls2002,
huangtingwei9988,
hzh0425,
ispobock,
mickqian,
xiezhq-hermann,
yctseng0211,
yhyang201,
yizhang2077 and
yuan-luo
as code owners
August 23, 2026 08:04
Contributor
Author
|
/tag-and-rerun-ci |
longxin9715
added a commit
to longxin9715/sglang
that referenced
this pull request
Aug 24, 2026
…n-transport1 * 'main' of https://github.com/sgl-project/sglang: (326 commits) [diffusion] feat: cache LoRA-merged weights in files the page cache can hold (sgl-project#36062) [diffusion] Speed up LingBot high-quality VAE decode (sgl-project#36024) [diffusion] Honor XDG cache for model overlays (sgl-project#36019) Support streaming session on NPU (sgl-project#32597) fix(xpu): read enable_deterministic_inference from the config bag (sgl-project#36149) xeon ci fail fast strategy change (sgl-project#36146) [diffusion] Fix Hunyuan QKV pack indexing at production video shapes (sgl-project#36009) [diffusion] Refresh quality and BCG benchmark skills (sgl-project#36016) [MoE] Gather the cutlass MoE activation and its scales in one launch (sgl-project#34915) [diffusion] feat: add plain component weight overrides (sgl-project#36086) [diffusion] feat: support loading mixed w4a8 text encoders (sgl-project#36037) [diffusion] Default Hunyuan VAE to tiled decode (sgl-project#36012) fix(xpu): enable compressed-tensors FP8 W8A8 on XPU (RedHatAI FP8-dynamic models) (sgl-project#33057) chore: move cuda_vmm_utils.py under srt/utils/ (sgl-project#36053) [Intel XPU] Add xpu pass for biased_topk and hash_topk (sgl-project#33323) [CPU] Fix NUMA/core binding for DP ranks (sgl-project#32856) [Fix] Harden FlashAttention CUDA graph metadata bounds (sgl-project#35454) [XPU] Use a fused GDN kernel from sgl-kernel for Qwen3.5 (sgl-project#33354) [diffusion] Fuse LongCat-Image QKNorm and interleaved RoPE (sgl-project#35995) [diffusion] Keep LongLive2 components resident on large GPUs (sgl-project#35993) ... # Conflicts: # python/sglang/srt/multimodal/processors/base_processor.py # python/sglang/srt/server_args.py
saturn-acc
pushed a commit
to saturn-acc/sglang
that referenced
this pull request
Aug 31, 2026
jakki-amd
pushed a commit
to jakki-amd/sglang
that referenced
this pull request
Sep 9, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Motivation
python/sglang/srt/cuda_vmm_utils.pysat at the top level ofsrt/, while its closest relatives —cuda_vmm_transport_utils.pyandcuda_ipc_transport_utils.py— already live undersrt/utils/. Move it next to them.Modifications
git mv python/sglang/srt/cuda_vmm_utils.py python/sglang/srt/utils/cuda_vmm_utils.py— the file contents are unchanged (rename shows 100% similarity).sglang.srt.cuda_vmm_utilstosglang.srt.utils.cuda_vmm_utils:srt/layers/moe/dwdp/{layout,page_pool,transport,weight_buffer}.pysrt/distributed/device_communicators/{custom_all_reduce_utils,custom_all_reduce_v2}.pysrt/model_executor/runner_utils/pool.pysrt/utils/cuda_vmm_transport_utils.pysrt/multimodal/transport/memory_pool.pysrt/mem_cache/kv_vmm_backing.pytest/registered/unit/test_cuda_vmm_utils.pyThe test file stays at
test/registered/unit/test_cuda_vmm_utils.py, consistent withtest_runai_utils.pyforsrt/utils/runai_utils.py.Checklist
pre-commit runclean on all touched files.grep -rn cuda_vmm_utilsconfirms no remaining references to the old path.CI States
Latest PR Test (Base): 🚫 Run #32627261656
Latest PR Test (Extra): ❌ Run #32627261597
Latest PR Test (AMD ROCm 7.2): ❌ Run #32627261680