未验证 提交 75fc4bf0 编写于 作者: L Leo Guo 提交者: GitHub

Fixed the bug in the api.cc file where there was an inconsistency between the...

Fixed the bug in the api.cc file where there was an inconsistency between the specified type (std::vector<DenseTensor*>&) in the function pointer kernel_signature and the type of the phi kernel parameter (std::vector<DenseTensor*>) when the phi kernel is set to output as std::vector<DenseTensor*>. test=kunlun (#54053)
上级 7e1dd338
...@@ -871,7 +871,7 @@ PADDLE_API {self.get_return_type(inplace_flag=True)} {api_func_name}({self.get_d ...@@ -871,7 +871,7 @@ PADDLE_API {self.get_return_type(inplace_flag=True)} {api_func_name}({self.get_d
} }
dense_out_trans_map = { dense_out_trans_map = {
'Tensor': 'phi::DenseTensor*', 'Tensor': 'phi::DenseTensor*',
'std::vector<Tensor>': 'std::vector<phi::DenseTensor*>&', 'std::vector<Tensor>': 'std::vector<phi::DenseTensor*>',
} }
sr_input_trans_map = { sr_input_trans_map = {
'const Tensor&': 'const phi::SelectedRows&', 'const Tensor&': 'const phi::SelectedRows&',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册