fix kernel selected error caused by alias name (#37984)

f68b175f · Chen Weihang · GitHub · 1f9a5d8f · f68b175f
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

paddle/fluid/framework/pten_utils.cc paddle/fluid/framework/pten_utils.cc +4 -2

未找到文件。
--- a/paddle/fluid/framework/pten_utils.cc
+++ b/paddle/fluid/framework/pten_utils.cc
@@ -15,6 +15,7 @@ limitations under the License. */
 #include <sstream>
 #include "paddle/fluid/framework/pten_utils.h"
+#include "paddle/pten/core/convert_utils.h"
 #include "paddle/pten/core/kernel_factory.h"
 #include "paddle/fluid/framework/lod_tensor.h"
@@ -190,8 +191,9 @@ KernelArgsNameMakerByOpProto::GetAttrsArgsNames() {
 }
 KernelSignature KernelArgsNameMakerByOpProto::GetKernelSignature() {
-  return KernelSignature(op_proto_->type(), GetInputArgsNames(),
+  return KernelSignature(pten::TransToPtenKernelName(op_proto_->type()),
-                         GetAttrsArgsNames(), GetOutputArgsNames());
+                         GetInputArgsNames(), GetAttrsArgsNames(),
+                         GetOutputArgsNames());
 }
 std::string KernelSignatureToString(const KernelSignature& signature) {