| 1436 | } |
| 1437 | |
| 1438 | void RdmaMemoryMgr::EvictMemoryRegion(void* addr, size_t length) { |
| 1439 | if (length == 0) return; |
| 1440 | mutex_lock l(mrs_mu_); |
| 1441 | auto iter = std::upper_bound(mrs_.begin(), mrs_.end(), addr, &Comparator); |
| 1442 | if (iter != std::end(mrs_) && iter->get()->addr == addr) { |
| 1443 | mrs_.erase(iter); |
| 1444 | RDMA_LOG(1) << "Evict memory region 0x" << std::hex << iter->get()->rkey; |
| 1445 | |
| 1446 | } else { |
| 1447 | LOG(WARNING) << "Failed to de-register memory region"; |
| 1448 | } |
| 1449 | } |
| 1450 | |
| 1451 | const TensorMetaData* RdmaMemoryMgr::GetTensorMetaData( |
| 1452 | const std::string& tensor_name) { |
no test coverage detected