/*! * Copyright (c) 2018 by Contributors * \file graph/network.h * \brief DGL networking related APIs */ #ifndef DGL_GRAPH_NETWORK_H_ #define DGL_GRAPH_NETWORK_H_ #include namespace dgl { namespace network { #define IS_SENDER true #define IS_RECEIVER false // TODO(chao): make these numbers configurable // Each single message cannot larger than 300 MB const int64_t kMaxBufferSize = 300 * 1024 * 2014; // Size of message queue is 1 GB const int64_t kQueueSize = 1024 * 1024 * 1024; // Maximal try count of connection const int kMaxTryCount = 500; // Control number const int CONTROL_NODEFLOW = 0; const int CONTROL_END_SIGNAL = 1; } // namespace network } // namespace dgl #endif // DGL_GRAPH_NETWORK_H_