From 75fc4bf06b2f35db9f3f3e030395ef157ddf14c8 Mon Sep 17 00:00:00 2001 From: Leo Guo <58431564+ZibinGuo@users.noreply.github.com> Date: Wed, 24 May 2023 17:40:05 +0800 Subject: [PATCH] Fixed the bug in the api.cc file where there was an inconsistency between the specified type (std::vector&) in the function pointer kernel_signature and the type of the phi kernel parameter (std::vector) when the phi kernel is set to output as std::vector. test=kunlun (#54053) --- paddle/phi/api/yaml/generator/api_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/phi/api/yaml/generator/api_base.py b/paddle/phi/api/yaml/generator/api_base.py index e4ac5726b9c..24f77bce383 100644 --- a/paddle/phi/api/yaml/generator/api_base.py +++ b/paddle/phi/api/yaml/generator/api_base.py @@ -871,7 +871,7 @@ PADDLE_API {self.get_return_type(inplace_flag=True)} {api_func_name}({self.get_d } dense_out_trans_map = { 'Tensor': 'phi::DenseTensor*', - 'std::vector': 'std::vector&', + 'std::vector': 'std::vector', } sr_input_trans_map = { 'const Tensor&': 'const phi::SelectedRows&', -- GitLab