fix kernel backend select bug (#41002)

1431305e · zyfncg · GitHub · d101334c · 1431305e
隐藏空白更改
内联并排

Showing with 12 addition and 9 deletion

paddle/phi/api/lib/kernel_dispatch.cc paddle/phi/api/lib/kernel_dispatch.cc +12 -9

未找到文件。
--- a/paddle/phi/api/lib/kernel_dispatch.cc
+++ b/paddle/phi/api/lib/kernel_dispatch.cc
@@ -25,16 +25,19 @@ namespace experimental {
 namespace detail {
 BackendSet GetTensorBackendSet(const phi::TensorBase& t) {
-  BackendSet backend_set(phi::TransToPhiBackend(t.place()));
+  if (t.initialized()) {
-  switch (t.layout()) {
+    BackendSet backend_set(phi::TransToPhiBackend(t.place()));
-    case DataLayout::MKLDNN:
+    switch (t.layout()) {
-      backend_set = backend_set | BackendSet(Backend::MKLDNN);
+      case DataLayout::MKLDNN:
-      break;
+        backend_set = backend_set | BackendSet(Backend::MKLDNN);
-    default:
+        break;
-      // do nothing
+      default:
-      break;
+        // do nothing
+        break;
+    }
+    return backend_set;
  }
-  return backend_set;
+  return BackendSet(Backend::UNDEFINED);
 }
 std::size_t CountLeadingZeros(uint64_t val) {