(self, metas: dict[int, MemoryBufferMetaList])
| 262 | return self._current_global_parameter_metas |
| 263 | |
| 264 | def load_metas(self, metas: dict[int, MemoryBufferMetaList]): |
| 265 | self._current_global_parameter_metas = metas |
| 266 | self._remote_rdma_devices = defaultdict(set) |
| 267 | for i, meta in self._current_global_parameter_metas.items(): |
| 268 | assert meta.rdma_device is not None, "meta.rdma_device should not be None" |
| 269 | assert meta.p2p_store_addr is not None, "meta.p2p_store_addr should not be None" |
| 270 | self._remote_rdma_devices[ |
| 271 | meta.rdma_device + "@" + meta.p2p_store_addr.split(":")[0] |
| 272 | ].add(i) |
| 273 | |
| 274 | def register_checkpoint( |
| 275 | self, |
no outgoing calls
no test coverage detected