MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / RdmaTensorRequest

Method RdmaTensorRequest

tensorflow/contrib/verbs/rdma.cc:1477–1492  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1475//*****************************************************************************
1476
1477RdmaTensorRequest::RdmaTensorRequest(
1478 uint32_t index, const string& key, int64 step_id, RdmaChannel* channel,
1479 Device* dst_dev, const Rendezvous::Args recv_args,
1480 const RdmaTensorRequest::RecvDoneCallback& done)
1481 : index_(index),
1482 key_(key),
1483 step_id_(step_id),
1484 channel_(channel),
1485 dst_dev_(dst_dev),
1486 recv_args_(recv_args),
1487 meta_data_(RdmaMemoryMgr::Singleton().GetTensorMetaData(key)),
1488 result_tensor_(nullptr),
1489 proxy_tensor_(nullptr),
1490 rdma_addr_(nullptr),
1491 mr_(nullptr),
1492 done_(done) {}
1493
1494RdmaTensorRequest::~RdmaTensorRequest() { DeallocateTensors(); }
1495

Callers

nothing calls this directly

Calls 1

GetTensorMetaDataMethod · 0.80

Tested by

no test coverage detected