From 9eff1a674fd4b5ae61d47f05274e8c18c5674a9a Mon Sep 17 00:00:00 2001 From: Jacek Czaja Date: Thu, 17 Dec 2020 08:14:07 +0100 Subject: [PATCH] Added missing format of oneDNN (#29670) --- paddle/fluid/platform/mkldnn_helper.h | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/paddle/fluid/platform/mkldnn_helper.h b/paddle/fluid/platform/mkldnn_helper.h index 2de08773df3..37747cd3fd3 100644 --- a/paddle/fluid/platform/mkldnn_helper.h +++ b/paddle/fluid/platform/mkldnn_helper.h @@ -277,6 +277,10 @@ inline mkldnn::memory::format_tag GetMKLDNNFormat( strides[3] >= strides[4] && strides[4] >= strides[1]) { return mkldnn::memory::format_tag::Acdeb8a; } + if (strides[0] >= strides[1] && strides[1] >= strides[2] && + strides[2] >= strides[3] && strides[3] >= strides[4]) { + return mkldnn::memory::format_tag::Abcde8a; + } } else if (inner_blks[0] == 8 && inner_idxs[0] == 1) { if (strides[0] >= strides[1] && strides[1] >= strides[2] && strides[2] >= strides[3] && strides[3] >= strides[4]) { -- GitLab