From 332082f25f6ce5fc4355fdec72cf80b1a3871f9c Mon Sep 17 00:00:00 2001 From: hebiao064 Date: Wed, 25 Jun 2025 05:55:00 +0000 Subject: [PATCH] Fix typo of flash_cache --- python/sglang/srt/managers/scheduler.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index 50f029cd32c..25a6d50e582 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -2201,8 +2201,8 @@ def update_weights_from_disk(self, recv_req: UpdateWeightFromDiskReqInput): """In-place update of the weights from disk.""" success, message = self.tp_worker.update_weights_from_disk(recv_req) if success: - flash_cache_success = self.flush_cache() - assert flash_cache_success, "Cache flush failed after updating weights" + flush_cache_success = self.flush_cache() + assert flush_cache_success, "Cache flush failed after updating weights" else: logger.error(message) return UpdateWeightFromDiskReqOutput(success, message, 0) @@ -2219,8 +2219,8 @@ def update_weights_from_distributed( """Update the online model parameter.""" success, message = self.tp_worker.update_weights_from_distributed(recv_req) if success: - flash_cache_success = self.flush_cache() - assert flash_cache_success, "Cache flush failed after updating weights" + flush_cache_success = self.flush_cache() + assert flush_cache_success, "Cache flush failed after updating weights" else: logger.error(message) return UpdateWeightsFromDistributedReqOutput(success, message) @@ -2231,8 +2231,8 @@ def update_weights_from_tensor(self, recv_req: UpdateWeightsFromTensorReqInput): # TODO extract common code b/t update_weights_from_distributed and update_weights_from_tensor later if success: if recv_req.flush_cache: - flash_cache_success = self.flush_cache() - assert flash_cache_success, "Cache flush failed after updating weights" + flush_cache_success = self.flush_cache() + assert flush_cache_success, "Cache flush failed after updating weights" else: logger.error(message) return UpdateWeightsFromTensorReqOutput(success, message)