diff --git a/src/plugin/impl/opr_footprint.cpp b/src/plugin/impl/opr_footprint.cpp index 2b691d875bb99d13891d765fc6c39d307f262530..cabb61c966433eebf2947f25e04c5fa22c589c17 100644 --- a/src/plugin/impl/opr_footprint.cpp +++ b/src/plugin/impl/opr_footprint.cpp @@ -127,7 +127,8 @@ uint64_t eval_conv_computation(const TensorShape& src_shape, src_shape[1] / group * 2; return hybird_nchwx ? computation : computation * 4; } - if (param.format == Param::Format::NCHW32) { + if (param.format == Param::Format::NCHW32 || + param.format == Param::Format::NCHW32_NCHW4) { return dst_shape.total_nr_elems() * fh * fw * src_shape[1] * 32 / group * 2; } @@ -157,11 +158,12 @@ uint64_t eval_conv_computation(const TensorShape& src_shape, }; if (param.format == Param::Format::NCHW4 || param.format == Param::Format::NCHW4_NCHW || - param.format == Param::Format::NCHW4_NCHW32 || + param.format == Param::Format::NCHW4_NCHW32 || param.format == Param::Format::NCHW88 || param.format == Param::Format::NCHW44 || param.format == Param::Format::NCHW44_DOT || - param.format == Param::Format::NCHW32) { + param.format == Param::Format::NCHW32 || + param.format == Param::Format::NCHW32_NCHW4) { return eval_conv_computation_nchwx(); } if (param.format == Param::Format::CHWN4) {