add comments, add to gpu not supported list

src/ngraph/ngraph.hpp

@@ -144,6 +144,7 @@
 #include "ngraph/op/power.hpp"
 #include "ngraph/op/product.hpp"
 #include "ngraph/op/quantize.hpp"
+#include "ngraph/op/recv.hpp"
 #include "ngraph/op/relu.hpp"
 #include "ngraph/op/replace_slice.hpp"
 #include "ngraph/op/reshape.hpp"
@@ -152,6 +153,7 @@
 #include "ngraph/op/scatter_add.hpp"
 #include "ngraph/op/scatter_nd_add.hpp"
 #include "ngraph/op/select.hpp"
+#include "ngraph/op/send.hpp"
 #include "ngraph/op/sigmoid.hpp"
 #include "ngraph/op/sign.hpp"
 #include "ngraph/op/sin.hpp"

src/ngraph/op/recv.hpp

@@ -30,6 +30,12 @@ namespace ngraph
             NGRAPH_API
             static const std::string type_name;
             const std::string& description() const override { return type_name; }
+            /// \brief Constructs an unitialized recv operation.
+            Recv();
+            /// \brief Constructs a Recv operation.
+            ///
+            /// \param arg The node for tensor to receive data
+            /// \param src_id the source id which could be rank or node id.
             Recv(const std::shared_ptr<Node>& arg, int src_id);
 
             void validate_and_infer_types() override;

src/ngraph/op/send.hpp

@@ -30,6 +30,12 @@ namespace ngraph
             NGRAPH_API
             static const std::string type_name;
             const std::string& description() const override { return type_name; }
+            /// \brief Constructs an unitialized send operation.
+            Send();
+            /// \brief Constructs a send operation.
+            ///
+            /// \param arg The node for input tensor
+            /// \param dest_id the target id which could be rank of node id.
             Send(const std::shared_ptr<Node>& arg, int dest_id);
 
             void validate_and_infer_types() override;

src/ngraph/runtime/gpu/gpu_backend.cpp

@@ -232,7 +232,9 @@ bool runtime::gpu::GPU_Backend::is_supported(const Node& op) const
                                    "GenerateMask",
                                    "DynBroadcast",
                                    "Transpose",
-                                   "Range"};
+                                   "Range",
+                                   "Recv",
+                                   "Send"};
 
     set<string> float_only = {"MaxPoolBackprop", "AvgPoolBackprop", "MaxPool", "Dot"};
 

src/ngraph/runtime/gpu/gpu_emitter.cpp

@@ -104,6 +104,7 @@
 #include "ngraph/op/power.hpp"
 #include "ngraph/op/product.hpp"
 #include "ngraph/op/quantize.hpp"
+#include "ngraph/op/recv.hpp"
 #include "ngraph/op/relu.hpp"
 #include "ngraph/op/replace_slice.hpp"
 #include "ngraph/op/reshape.hpp"
@@ -113,6 +114,7 @@
 #include "ngraph/op/scatter_add.hpp"
 #include "ngraph/op/scatter_nd_add.hpp"
 #include "ngraph/op/select.hpp"
+#include "ngraph/op/send.hpp"
 #include "ngraph/op/sigmoid.hpp"
 #include "ngraph/op/sign.hpp"
 #include "ngraph/op/sin.hpp"