PaddlePaddle
diff --git a/‎paddle/fluid/operators/collective/c_scatter_op.cc
Copy file name to clipboard
-96Lines changed: 0 additions & 96 deletions b/‎paddle/fluid/operators/collective/c_scatter_op.cc
Copy file name to clipboard
-96Lines changed: 0 additions & 96 deletions
diff --git a/‎paddle/fluid/operators/collective/c_scatter_op.cu.cc
Copy file name to clipboard
-176Lines changed: 0 additions & 176 deletions b/‎paddle/fluid/operators/collective/c_scatter_op.cu.cc
Copy file name to clipboard
-176Lines changed: 0 additions & 176 deletions
diff --git a/‎paddle/fluid/operators/collective/c_scatter_op.h
Copy file name to clipboard
-58Lines changed: 0 additions & 58 deletions b/‎paddle/fluid/operators/collective/c_scatter_op.h
Copy file name to clipboard
-58Lines changed: 0 additions & 58 deletions
diff --git a/‎paddle/phi/infermeta/unary.cc
Copy file name to clipboard
+21-1Lines changed: 21 additions & 1 deletion b/‎paddle/phi/infermeta/unary.cc
Copy file name to clipboard
+21-1Lines changed: 21 additions & 1 deletion
diff --git a/‎paddle/phi/infermeta/unary.h
Copy file name to clipboard
+2-1Lines changed: 2 additions & 1 deletion b/‎paddle/phi/infermeta/unary.h
Copy file name to clipboard
+2-1Lines changed: 2 additions & 1 deletion
@@ -795,7 +795,27 @@ void CropInferMeta(const MetaTensor& x,
   out->set_dtype(x.dtype());
 }
 
-void CScatterInferMeta(const MetaTensor& x, int nranks, MetaTensor* out) {
+void CScatterInferMeta(const MetaTensor& x,
+                       int ring_id,
+                       int root_id,
+                       int nranks,
+                       MetaTensor* out) {
+  PADDLE_ENFORCE_GE(nranks,
+                    2,
+                    common::errors::InvalidArgument(
+                        "The number of ranks (%d) must be greater than 1 "
+                        "to use collective op (c_scatter op).",
+                        nranks));
+  PADDLE_ENFORCE_GE(
+      root_id,
+      0,
+      common::errors::InvalidArgument(
+          "The root_id (%d) for c_scatter_op must be non-negative.", root_id));
+  PADDLE_ENFORCE_GE(
+      ring_id,
+      0,
+      common::errors::InvalidArgument(
+          "The ring_id (%d) for c_scatter_op must be non-negative.", ring_id));
   auto dim = x.dims();
   dim[0] = dim[0] / nranks;
   if (dim[0] < 0) dim[0] = -1;
 
@@ -158,7 +158,8 @@ void CropInferMeta(const MetaTensor& x,
                    MetaTensor* out,
                    MetaConfig config = MetaConfig());
 
-void CScatterInferMeta(const MetaTensor& x, int nranks, MetaTensor* out);
+void CScatterInferMeta(
+    const MetaTensor& x, int ring_id, int root, int nranks, MetaTensor* out);
 
 void CSplitInferMeta(const MetaTensor& x, int nranks, MetaTensor* out);