[Pir] Add ReduceScatter Yaml Definition (#58991)

JZ-LIANG · web-flow · commit 431a0d53bd15 · 2023-11-16T20:24:15.000+08:00
* add reducescatter for pir yaml

* adopt for static op

* remove list
diff --git a/paddle/fluid/pir/dialect/op_generator/ops_api_gen.py b/paddle/fluid/pir/dialect/op_generator/ops_api_gen.py
@@ -105,6 +105,7 @@
     'c_embedding',
     'c_identity',
     'c_reduce_sum',
+    'c_reducescatter',
     'dpsgd',
     'embedding_grad_sparse',
     'fused_batch_norm_act_',
diff --git a/paddle/fluid/pir/dialect/operator/ir/ops.yaml b/paddle/fluid/pir/dialect/operator/ir/ops.yaml
@@ -38,6 +38,16 @@
     backend: place>
     data_type : dtype
 
+- op : c_reducescatter
+  args : (Tensor x, int ring_id = 0, int nranks = 1, bool use_calc_stream = false)
+  output : Tensor(out)
+  infer_meta :
+    func : ReduceScatterInferMeta
+    param: [x, nranks]
+  kernel :
+    func : reduce_scatter
+    param: [x, nranks]
+
 - op : embedding_grad_sparse
   args : (Tensor x, Tensor weight, Tensor out_grad, int64_t padding_idx = -1, bool sparse = false)
   output : SelectedRows(weight_grad)
