Browse Source

fix(dnn): keep consistent limit between deduce and compute

GitOrigin-RevId: 8de5f17ced
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
0307598a80
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      dnn/src/fallback/conv_bias/winograd/winograd.h

+ 5
- 0
dnn/src/fallback/conv_bias/winograd/winograd.h View File

@@ -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;


Loading…
Cancel
Save