From 7a8524b444bdaf483faab8af75059fdb0d42a5c7 Mon Sep 17 00:00:00 2001 From: Mayyyy <30303931+hustmf@users.noreply.github.com> Date: Thu, 13 Nov 2025 11:03:32 +0800 Subject: [PATCH] [Ascend] add npu synchronize (#13154) Co-authored-by: may_feimei --- python/sglang/srt/managers/scheduler_update_weights_mixin.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/managers/scheduler_update_weights_mixin.py b/python/sglang/srt/managers/scheduler_update_weights_mixin.py index 9bed7030d..fafdd48f4 100644 --- a/python/sglang/srt/managers/scheduler_update_weights_mixin.py +++ b/python/sglang/srt/managers/scheduler_update_weights_mixin.py @@ -132,7 +132,7 @@ class SchedulerUpdateWeightsMixin: if GPU_MEMORY_TYPE_CUDA_GRAPH in tags: self.memory_saver_adapter.pause(GPU_MEMORY_TYPE_CUDA_GRAPH) - torch.cuda.synchronize() + torch.get_device_module().synchronize() return ReleaseMemoryOccupationReqOutput()