(self, room: int)
| 320 | torch.cuda.empty_cache() |
| 321 | |
| 322 | def remove(self, room: int): |
| 323 | self.release_memory(room) |
| 324 | |
| 325 | self.data_receiver.pop(room, None) |
| 326 | self._data_mgr_sidecar.unwatch_input(room) |
| 327 | |
| 328 | if self.data_mgr is None: |
| 329 | return |
| 330 | |
| 331 | self.data_mgr.remove(room) |
| 332 | |
| 333 | def release(self): |
| 334 | for room in list(self._rdma_buffers.keys()): |
no test coverage detected