PaddlePaddle
diff --git a/‎paddle/fluid/operators/beam_search_decode_op.cc
Copy file name to clipboard
-119Lines changed: 0 additions & 119 deletions b/‎paddle/fluid/operators/beam_search_decode_op.cc
Copy file name to clipboard
-119Lines changed: 0 additions & 119 deletions
diff --git a/‎paddle/fluid/operators/beam_search_decode_op.cu.cc
Copy file name to clipboard
-28Lines changed: 0 additions & 28 deletions b/‎paddle/fluid/operators/beam_search_decode_op.cu.cc
Copy file name to clipboard
-28Lines changed: 0 additions & 28 deletions
diff --git a/‎paddle/fluid/operators/beam_search_decode_op.h
Copy file name to clipboard
-52Lines changed: 0 additions & 52 deletions b/‎paddle/fluid/operators/beam_search_decode_op.h
Copy file name to clipboard
-52Lines changed: 0 additions & 52 deletions
diff --git a/‎paddle/fluid/operators/beam_search_decode_op_xpu.cc
Copy file name to clipboard
-124Lines changed: 0 additions & 124 deletions b/‎paddle/fluid/operators/beam_search_decode_op_xpu.cc
Copy file name to clipboard
-124Lines changed: 0 additions & 124 deletions
diff --git a/‎paddle/fluid/pir/dialect/op_generator/ops_api_gen.py
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions b/‎paddle/fluid/pir/dialect/op_generator/ops_api_gen.py
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions
diff --git a/‎paddle/phi/infermeta/binary.cc
Copy file name to clipboard
+8Lines changed: 8 additions & 0 deletions b/‎paddle/phi/infermeta/binary.cc
Copy file name to clipboard
+8Lines changed: 8 additions & 0 deletions
diff --git a/‎paddle/phi/infermeta/binary.h
Copy file name to clipboard
+8Lines changed: 8 additions & 0 deletions b/‎paddle/phi/infermeta/binary.h
Copy file name to clipboard
+8Lines changed: 8 additions & 0 deletions
@@ -113,57 +113,5 @@ struct BeamSearchDecodeFunctor {
   phi::DenseTensor* score_tensor_;
 };
 
-template <typename T, typename DeviceContext>
-class BeamSearchDecodeOpKernel : public framework::OpKernel<T> {
- public:
-  void Compute(const framework::ExecutionContext& context) const override {
-    const LoDTensorArray* ids = context.Input<LoDTensorArray>("Ids");
-    const LoDTensorArray* scores = context.Input<LoDTensorArray>("Scores");
-    const size_t step_num = ids->size();
-    PADDLE_ENFORCE_GT(
-        step_num,
-        0UL,
-        phi::errors::InvalidArgument(
-            "beam search steps, which is the"
-            "size of Input(Ids) LoDTensorArray. beam search steps should "
-            "be larger than 0, but received %d. ",
-            step_num));
-    const size_t source_num = ids->at(0).lod().at(0).size() - 1;
-    PADDLE_ENFORCE_GT(
-        source_num,
-        0UL,
-        phi::errors::InvalidArgument(
-            "source_num is the sequence number of the"
-            "first decoding step, indicating by Input(Ids)[0].lod[0].size. "
-            "The number of source_num should be larger than"
-            "0, but received %d. ",
-            source_num));
-
-    for (size_t i = 0; i < step_num; ++i) {
-      PADDLE_ENFORCE_EQ(
-          ids->at(i).lod().size(),
-          2UL,
-          phi::errors::InvalidArgument(
-              "For the i step in beam search steps,"
-              "the size of Input(Ids)[i].lod() should larger than 2,"
-              "but received %d. ",
-              ids->at(i).lod().size()));
-    }
-
-    size_t beam_size = context.Attr<int>("beam_size");
-    int end_id = context.Attr<int>("end_id");
-
-    // prepare output
-    phi::DenseTensor* sentenceIds =
-        context.Output<phi::DenseTensor>("SentenceIds");
-    phi::DenseTensor* sentenceScores =
-        context.Output<phi::DenseTensor>("SentenceScores");
-
-    BeamSearchDecodeFunctor bs(
-        *ids, *scores, sentenceIds, sentenceScores, beam_size, end_id);
-    bs.apply_mix<T>();
-  }
-};
-
 }  // namespace operators
 }  // namespace paddle
@@ -148,6 +148,7 @@
     'c_allgather',
     'qkv_unpack_mha',
     'hash',
+    'beam_search_decode',
 ]
 
 NO_NEED_GEN_STATIC_ONLY_APIS = [
 
@@ -256,6 +256,14 @@ void BCELossInferMeta(const MetaTensor& input,
   out->share_lod(input);
 }
 
+void BeamSearchDecodeInferMeta(const MetaTensor& ids,
+                               const MetaTensor& scores,
+                               int beam_size,
+                               int end_id,
+                               MetaTensor* sentence_ids,
+                               MetaTensor* sentence_scores,
+                               MetaConfig config) {}
+
 void BincountInferMeta(const MetaTensor& x,
                        const MetaTensor& weights,
                        const Scalar& minlength,
 
@@ -63,6 +63,14 @@ void BCELossInferMeta(const MetaTensor& input,
                       MetaTensor* out,
                       MetaConfig config = MetaConfig());
 
+void BeamSearchDecodeInferMeta(const MetaTensor& ids,
+                               const MetaTensor& scores,
+                               int beam_size,
+                               int end_id,
+                               MetaTensor* sentence_ids,
+                               MetaTensor* sentence_scores,
+                               MetaConfig config = MetaConfig());
+
 void BincountInferMeta(const MetaTensor& x,
                        const MetaTensor& weights,
                        const Scalar& minlength,
Original file line number	Diff line number	Diff line change
`@@ -148,6 +148,7 @@`
`148`	`148`	`'c_allgather',`
`149`	`149`	`'qkv_unpack_mha',`
`150`	`150`	`'hash',`
	`151`	`+ 'beam_search_decode',`
`151`	`152`	`]`
`152`	`153`
`153`	`154`	`NO_NEED_GEN_STATIC_ONLY_APIS = [`