Releases the RDMA buffers and clears GPU cache.
(self, room: int)
| 692 | """ |
| 693 | Releases the RDMA buffers and clears GPU cache. |
| 694 | """ |
| 695 | if room in self._rdma_buffers: |
| 696 | self._rdma_buffers.pop(room, None) |
| 697 | torch.cuda.empty_cache() |
| 698 | |
| 699 | def remove(self, room: int): |
| 700 | self.release_memory(room) |
| 701 | |
| 702 | self.data_sender.pop(room, None) |
| 703 | self._data_mgr_sidecar.unwatch_output(room) |