| 979 | } |
| 980 | |
| 981 | RdmaTensorResponse* RdmaChannel::UpdateTensorResponse(const RdmaMessage& rm) { |
| 982 | mutex_lock lock{mu_}; |
| 983 | auto it = responses_table_.find(rm.request_index_); |
| 984 | CHECK(it != responses_table_.end()) << "No response found."; |
| 985 | RdmaTensorResponse* response = &it->second; |
| 986 | response->Update(rm); |
| 987 | return response; |
| 988 | } |
| 989 | |
| 990 | void RdmaChannel::RemoveTensorResponse(uint32_t request_index) { |
| 991 | mutex_lock lock{mu_}; |
no test coverage detected