| 1606 | } |
| 1607 | |
| 1608 | void RdmaTensorRequest::RecvTensorMetaData(DataType dtype, TensorShape shape, |
| 1609 | bool is_dead, size_t proto_size) { |
| 1610 | meta_data_ = RdmaMemoryMgr::Singleton().SetTensorMetaData( |
| 1611 | key_, dtype, shape, is_dead, proto_size); |
| 1612 | |
| 1613 | DeallocateTensors(); |
| 1614 | AllocateTensorsAsync( |
| 1615 | [this](const Status& s) { Send(RDMA_MESSAGE_TENSOR_RE_REQUEST); }); |
| 1616 | } |
| 1617 | |
| 1618 | void RdmaTensorRequest::RecvTensorContent() { |
| 1619 | bool can_memcpy = DataTypeCanUseMemcpy(meta_data_->data_type_); |
no test coverage detected