[RL] [KVCache] let cache transfer managers update key prefix after weight update and add unit tests (#7083)

* [test] add a few unit tests

* [feat] update key prefix when model weights are updated

* [test] try to fix test_worker_process
This commit is contained in:
Yonghua Li
2026-04-02 19:58:41 +08:00
committed by GitHub
parent 9f3b3ce7f5
commit 98f3fc9267
8 changed files with 636 additions and 11 deletions
@@ -1098,6 +1098,14 @@ class CacheTransferManager:
logger.info("✅ Successfully resumed transfer")
return True
def _handle_update_weights(self):
if self.storage_backend_type is not None:
self._update_key_prefix()
logger.info("✅ Successfully updated cache key prefix after weight update")
else:
logger.info("💡 Cache storage backend is disabled, skip updating cache key prefix")
return True
def _handle_sleep(self):
if self.is_sleeping:
logger.info("💡 Cache transfer manager is already sleeping, no need to sleep again!")
@@ -1128,6 +1136,7 @@ class CacheTransferManager:
handlers = {
"pause": self._handle_pause,
"resume": self._handle_resume,
"update_weights": self._handle_update_weights,
"sleep": self._handle_sleep,
"wakeup": self._handle_wakeup,
}