【PIR OpTest Fix No.31】 fix test_number_count_op (PaddlePaddle#60055)

* add numbercount

* add number_count to NEED_GEN_STATIC_ONLY_APIS

* revert ops_api_gen

* add NEED_GEN_STATIC_ONLY_APIS to number_count

* numbers => Numbers

* numbers => Numbers in py

* numbers => Numbers

* fix data_type err && Numbers|numbers err

* Number revert

Author: DrRyanHuang

paddle/fluid/operators/number_count_op.cc

@@ -46,7 +46,7 @@ class NumberCountOpMaker : public framework::OpProtoAndCheckerMaker {
   void Make() override {
     AddInput("numbers", "(Tensor) The input gate index tensor.");
     AddOutput("Out", "(Tensor) The output number count tensor.");
-    AddAttr<int>("upper_range", "（int), The number of different numbers.");
+    AddAttr<int>("upper_range", "(int), The number of different numbers.");
 
     AddComment(R"DOC(number_count Operator.count numbers.)DOC");
   }

paddle/fluid/pir/dialect/op_generator/ops_api_gen.py

@@ -101,6 +101,7 @@
     'get_tensor_from_selected_rows',
     'print',
     'sequence_mask',
+    'number_count',
 ]
 
 NO_NEED_GEN_STATIC_ONLY_APIS = [

paddle/fluid/pir/dialect/operator/ir/ops.yaml

@@ -1428,6 +1428,15 @@
   optional: dropout1_seed, dropout2_seed, linear1_bias, linear2_bias, ln1_scale, ln1_bias, ln2_scale, ln2_bias, ln2_mean, ln2_variance, ln1_mean, ln1_variance, ln1_out
   backward: fused_feedforward_grad
 
+- op: number_count
+  args: (Tensor numbers, int upper_range)
+  output: Tensor(out)
+  infer_meta:
+     func: NumberCountInferMeta
+  kernel:
+     func: number_count
+     data_type: numbers
+
 - op: sparse_momentum
   args: (Tensor param, Tensor grad, Tensor velocity, Tensor index, Tensor learning_rate, Tensor master_param,float mu, Scalar axis=0, bool use_nesterov=false,str regularization_method="", float regularization_coeff=0.0f, bool multi_precision=false, float rescale_grad=1.0f)
   output: Tensor(param_out), Tensor(velocity_out), Tensor(master_param_out)

paddle/phi/api/yaml/op_compat.yaml

@@ -3472,6 +3472,12 @@
   outputs :
     out : Out
 
+- op: number_count
+  inputs :
+    {numbers: numbers}
+  outputs :
+    out : Out
+
 - op: read_from_array
   inputs:
     array : X

paddle/phi/infermeta/unary.cc

@@ -5313,6 +5313,12 @@ void StridedUnChangedInferMeta(const MetaTensor& x, MetaTensor* out) {
   out->set_strides(x.strides());
 }
 
+void NumberCountInferMeta(const MetaTensor& x,
+                          int upper_range,
+                          MetaTensor* out) {
+  out->share_meta(x);
+}
+
 }  // namespace phi
 
 PD_REGISTER_INFER_META_FN(flatten, phi::FlattenInferMeta);

paddle/phi/infermeta/unary.h

@@ -789,6 +789,10 @@ void UnStackInferMeta(const MetaTensor& x,
                       int num,
                       std::vector<MetaTensor*> outs);
 
+void NumberCountInferMeta(const MetaTensor& x,
+                          int upper_range,
+                          MetaTensor* out);
+
 void StridedUnChangedInferMeta(const MetaTensor& x, MetaTensor* out);
 
 }  // namespace phi

test/white_list/pir_op_test_white_list

@@ -219,6 +219,7 @@ test_nms_op
 test_nn_functional_hot_op
 test_nonzero_api
 test_norm_op
+test_number_count_op
 test_numel_op
 test_one_hot_v2_op
 test_one_hot_v2_op_static_build