diff options
Diffstat (limited to 'tensorflow/contrib/verbs/rdma_mgr.h')
| -rw-r--r-- | tensorflow/contrib/verbs/rdma_mgr.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/verbs/rdma_mgr.h b/tensorflow/contrib/verbs/rdma_mgr.h index e711e60478..29227a9544 100644 --- a/tensorflow/contrib/verbs/rdma_mgr.h +++ b/tensorflow/contrib/verbs/rdma_mgr.h @@ -38,6 +38,7 @@ class RdmaMgr { RdmaChannel* FindChannel(const string& key); void SetupChannels(); bool ConnectivityCheck(); + void InitAllocators(); const string& local_worker() { return local_worker_; } private: |