From ba69c153f66306fc7972521eb7a4bda794e1dde0 Mon Sep 17 00:00:00 2001 From: Nan Jiang <59716405+nanjiangwill@users.noreply.github.com> Date: Sun, 6 Jul 2025 16:51:29 -0700 Subject: [PATCH] [RL]: Fix error tagging in multi-stage wake up (#7812) Co-authored-by: hebiao064 --- python/sglang/srt/managers/scheduler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index 3ff6eaf57..7dd17793a 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -2333,9 +2333,8 @@ class Scheduler( def release_memory_occupation(self, recv_req: ReleaseMemoryOccupationReqInput): tags = recv_req.tags - import subprocess - if tags is None: + if tags is None or len(tags) == 0: tags = [GPU_MEMORY_TYPE_WEIGHTS, GPU_MEMORY_TYPE_KV_CACHE] if GPU_MEMORY_TYPE_KV_CACHE in tags: @@ -2353,6 +2352,7 @@ class Scheduler( def resume_memory_occupation(self, recv_req: ResumeMemoryOccupationReqInput): tags = recv_req.tags + if tags is None or len(tags) == 0: tags = [GPU_MEMORY_TYPE_WEIGHTS, GPU_MEMORY_TYPE_KV_CACHE]