[Ascend] add npu synchronize (#13154)

Co-authored-by: may_feimei <meifei5@huawei.com>
2025-11-13 11:03:32 +08:00
parent 909d0d382f
commit 7a8524b444
1 changed files with 1 additions and 1 deletions
--- 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()