提交 0307598a 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: 8de5f17ced98f4a4200618a42291baeec210f654
上级 354b478c
...@@ -214,6 +214,11 @@ public: ...@@ -214,6 +214,11 @@ public:
SmallVector<TensorLayout> deduce_preprocessed_filter_layout( SmallVector<TensorLayout> deduce_preprocessed_filter_layout(
const NCBKernSizeParam& param, fallback::MatrixMulImpl::AlgoBase*) { 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 OC = param.filter_meta.ocpg;
size_t IC = param.filter_meta.icpg; size_t IC = param.filter_meta.icpg;
size_t GROUP = param.filter_meta.group; size_t GROUP = param.filter_meta.group;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册