From 0307598a8028c14ab0747a08b84928bb46cdbc13 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 14 Sep 2020 18:30:46 +0800
Subject: [PATCH] fix(dnn): keep consistent limit between deduce and compute

GitOrigin-RevId: 8de5f17ced98f4a4200618a42291baeec210f654
---
 dnn/src/fallback/conv_bias/winograd/winograd.h | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/dnn/src/fallback/conv_bias/winograd/winograd.h b/dnn/src/fallback/conv_bias/winograd/winograd.h
index eb1796ee..fda01133 100644
--- a/dnn/src/fallback/conv_bias/winograd/winograd.h
+++ b/dnn/src/fallback/conv_bias/winograd/winograd.h
@@ -214,6 +214,11 @@ public:
 
     SmallVector<TensorLayout> deduce_preprocessed_filter_layout(
             const NCBKernSizeParam& param, fallback::MatrixMulImpl::AlgoBase*) {
+        if (param.filter_meta.format != param::ConvBias::Format::NCHW &&
+            param.filter_meta.format != param::ConvBias::Format::NCHW88 &&
+            param.filter_meta.format != param::ConvBias::Format::NCHW44) {
+            return {};
+        }
         size_t OC = param.filter_meta.ocpg;
         size_t IC = param.filter_meta.icpg;
         size_t GROUP = param.filter_meta.group;