fix mkldnn bugs (#45770)

23def396 · YuanRisheng · GitHub · 192b3033 · 23def396
显示空白变更内容
内联并排

Showing with 4 addition and 2 deletion

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

未找到文件。
--- a/paddle/fluid/framework/operator.cc
+++ b/paddle/fluid/framework/operator.cc
@@ -1328,8 +1328,10 @@ bool OperatorWithKernel::SupportsMKLDNN(
  auto has_phi_kernel =
      std::any_of(phi_kernels.begin(),
                  phi_kernels.end(),
-                  [](phi::KernelKeyMap::const_reference kern_pair) {
+                  [data_type](phi::KernelKeyMap::const_reference kern_pair) {
-                    return kern_pair.first.backend() == phi::Backend::ONEDNN;
+                    return kern_pair.first.backend() == phi::Backend::ONEDNN &&
+                           kern_pair.first.dtype() ==
+                               framework::TransToPhiDataType(data_type);
                  });
  if (has_phi_kernel) {
    return true;