Added missing oneDNN format (#25450)

test=develop

Added missing oneDNN format (#25450)
test=develop
a5d1592f · Jacek Czaja · GitHub · 172d4ecb · a5d1592f
显示空白变更内容
内联并排

Showing with 3 addition and 0 deletion

paddle/fluid/platform/mkldnn_helper.h paddle/fluid/platform/mkldnn_helper.h +3 -0

未找到文件。
--- a/paddle/fluid/platform/mkldnn_helper.h
+++ b/paddle/fluid/platform/mkldnn_helper.h
@@ -178,6 +178,9 @@ inline mkldnn::memory::format_tag GetMKLDNNFormat(
      if (strides[0] >= strides[1] && strides[1] >= strides[2] &&
          strides[2] >= strides[3]) {
        return mkldnn::memory::format_tag::nchw;
+      } else if (strides[2] >= strides[3] && strides[3] >= strides[1] &&
+                 strides[1] >= strides[0]) {
+        return mkldnn::memory::format_tag::cdba;
      } else {
        return mkldnn::memory::format_tag::nhwc;
      }