Update cache_messager.py (#3172)

This commit is contained in:
ltd0924
2025-08-04 14:32:34 +08:00
committed by GitHub
parent 4021d66ea5
commit c9e6ce1518

View File

@@ -142,7 +142,7 @@ class CacheMessager:
self.gpu_id = gpu_id
self.cache_info = dict()
self.dp_rank_id = local_data_parallel_id
self.dp_rank_id = self.rank + local_data_parallel_id * self.nranks
layerwise_send_cache_thread = threading.Thread(target=self._prefill_layerwise_send_cache_thread)
layerwise_send_cache_thread.daemon = True