| 970 | #endif // GOOGLE_CUDA |
| 971 | |
| 972 | RdmaTensorResponse* RdmaChannel::AddTensorResponse(const RdmaMessage& rm) { |
| 973 | mutex_lock lock{mu_}; |
| 974 | auto it = |
| 975 | responses_table_.emplace(rm.request_index_, RdmaTensorResponse(this, rm)); |
| 976 | CHECK(it.second) << "Response with the ID " << rm.request_index_ |
| 977 | << " already exists."; |
| 978 | return &it.first->second; |
| 979 | } |
| 980 | |
| 981 | RdmaTensorResponse* RdmaChannel::UpdateTensorResponse(const RdmaMessage& rm) { |
| 982 | mutex_lock lock{mu_}; |
no test coverage detected