MegEngine/dnn/src/arm_common/conv_bias/int8/direct_dotprod.cpp

#include "src/arm_common/conv_bias/int8/direct_dotprod.h"
#if MGB_ENABLE_DOT
#include "src/arm_common/elemwise_helper/elemwise_op.h"
#include "src/arm_common/simd_macro/marm_neon.h"
#include "src/common/utils.h"
#include "src/fallback/conv_bias/common.h"

using namespace megdnn;
using namespace arm_common;
using megdnn::arm_common::ReluOp;
using megdnn::arm_common::TypeCvtOp;

inline int8x16_t vqtbl1q_s8_v7(int8x16_t a, uint8x16_t index) {
    int8x8x2_t src;
    src.val[0] = vget_low_s8(a);
    src.val[1] = vget_high_s8(a);
    uint8x8_t index_low = vget_low_u8(index);
    uint8x8_t index_high = vget_high_u8(index);
    int8x8_t r00 = vtbl2_s8(src, vreinterpret_s8_u8(index_low));
    int8x8_t r01 = vtbl2_s8(src, vreinterpret_s8_u8(index_high));
    int8x16_t r = vcombine_s8(r00, r01);
    return r;
}

#define ST1_S32X4(dst0, tptr) vst1q_s32(tptr, dst0);

#define ST2_S32X4X2(dst0, tptr) vst2q_s32(tptr, dst0);

#define POSTPROCESS_1X8(dst0, dst1, tptr, dptr)                \
    if (last_ic) {                                             \
        op({{dst0, dst1}}, reinterpret_cast<dt_qint8*>(dptr)); \
    } else {                                                   \
        ST1_S32X4(dst0, tptr);                                 \
        ST1_S32X4(dst1, tptr + 4);                             \
    }

#define POSTPROCESS2_1X8(dst0, tptr, dptr)                         \
    if (last_ic) {                                                 \
        int32x4x2_t temp;                                          \
        int32x4_t temp00, temp11;                                  \
        temp = vzipq_s32(dst0.val[0], dst0.val[1]);                \
        temp00 = temp.val[0];                                      \
        temp11 = temp.val[1];                                      \
        op({{temp00, temp11}}, reinterpret_cast<dt_qint8*>(dptr)); \
    } else {                                                       \
        ST2_S32X4X2(dst0, tptr);                                   \
    }

#define POSTPROCESS_2X4(dst0, dst1, tptr1, tptr2, dptr1, dptr2)          \
    if (last_ic) {                                                       \
        int32x2_t res = reinterpret_cast<int32x2_t>(op({{dst0, dst1}})); \
        vst1_lane_s32(reinterpret_cast<int32_t*>(dptr1), res, 0);        \
        vst1_lane_s32(reinterpret_cast<int32_t*>(dptr2), res, 1);        \
    } else {                                                             \
        ST1_S32X4(dst0, tptr1);                                          \
        ST1_S32X4(dst1, tptr2);                                          \
    }

#define POSTPROCESS_1X4(dst0, tptr, dptr)                                \
    if (last_ic) {                                                       \
        int32x4_t dst1 = vdupq_n_s32(0);                                 \
        int32x2_t res = reinterpret_cast<int32x2_t>(op({{dst0, dst1}})); \
        vst1_lane_s32(reinterpret_cast<int32_t*>(dptr), res, 0);         \
    } else {                                                             \
        ST1_S32X4(dst0, tptr);                                           \
    }

#define CALC_0(_k_idx, _c_idx)                 \
    _elem = vqtbl1q_s8_v7(_tmp, _idx##_c_idx); \
    _sum0##_c_idx = vdotq_s32(_sum0##_c_idx, _k##_k_idx, _elem);

#define CALC_1(_k_idx, _c_idx)                 \
    _elem = vqtbl1q_s8_v7(_tmp, _idx##_c_idx); \
    _sum1##_c_idx = vdotq_s32(_sum1##_c_idx, _k##_k_idx, _elem);

#define CALC_2(_k1_idx, _k2_idx, _c_idx)                          \
    _elem = vqtbl1q_s8_v7(_tmp, _idx##_c_idx);                    \
    _sum0##_c_idx = vdotq_s32(_sum0##_c_idx, _k##_k1_idx, _elem); \
    _sum1##_c_idx = vdotq_s32(_sum1##_c_idx, _k##_k2_idx, _elem);

template <bool first_ic, bool last_ic, BiasMode bias_mode, typename Op>
MEGDNN_ATTRIBUTE_TARGET("dotprod")
void conv_bias::conv_direct_stride1_2x2_int8_dot(
        const int8_t* src, const int8_t* filter, const int32_t* bias, int32_t* temp,
        int8_t* dst, const size_t IH, const size_t IW, const size_t OH, const size_t OW,
        const Op& op) {
    MEGDNN_MARK_USED_VAR(IH);
    const size_t tail_step = IW - OW;
    const uint8x16_t _idx0 = {0, 1, 16, 16, 1, 2, 16, 16, 2, 3, 16, 16, 3, 4, 16, 16};
    const uint8x16_t _idx1 = {4, 5, 16, 16, 5, 6, 16, 16, 6, 7, 16, 16, 7, 8, 16, 16};
    int32_t* outptr = temp;
    int32_t* outptr2 = outptr + OW;
    int8_t* dstptr = dst;
    int8_t* dstptr2 = dstptr + OW;
    const int32_t* __restrict bptr = bias;

    const int8_t* r0 = src;
    const int8_t* r1 = src + IW;
    const int8_t* r2 = src + 2 * IW;

    const int8_t* k0 = filter;

    int8x16_t _k =
            vreinterpretq_s8_s32(vdupq_n_s32(*reinterpret_cast<const int32_t*>(k0)));
    uint8x16_t _idx = {0, 1, 16, 16, 0, 1, 16, 16, 0, 1, 16, 16, 0, 1, 16, 16};
    int8x16_t _k1 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {2, 3, 16, 16, 2, 3, 16, 16, 2, 3, 16, 16, 2, 3, 16, 16};
    int8x16_t _k23 = vqtbl1q_s8_v7(_k, _idx);

    int8x16_t _tmp, _elem;
    const int width = OW >> 2;
    size_t h = 0;
    for (; h + 1 < OH; h += 2) {
        int w = 0;
        for (; w + 2 < width; w += 2) {
            int32x4_t _sum00, _sum01, _sum10, _sum11;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
                _sum10 = vld1q_s32(outptr2);
                _sum11 = vld1q_s32(outptr2 + 4);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                    _sum10 = _sum00;
                    _sum11 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                    _sum11 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(1, 0);
            CALC_0(1, 1);

            _tmp = vld1q_s8(r1);
            CALC_2(23, 1, 0);
            CALC_2(23, 1, 1);

            _tmp = vld1q_s8(r2);
            CALC_1(23, 0);
            CALC_1(23, 1);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);
            POSTPROCESS_1X8(_sum10, _sum11, outptr2, dstptr2);

            r0 += 8;
            r1 += 8;
            r2 += 8;
            outptr += 8;
            outptr2 += 8;
            dstptr += 8;
            dstptr2 += 8;
        }

        for (; w < width; w++) {
            int32x4_t _sum00, _sum10;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum10 = vld1q_s32(outptr2);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum10 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                }
            }

            _tmp = vtranslq_s8(vld1_s8(r0));
            CALC_0(1, 0);

            _tmp = vtranslq_s8(vld1_s8(r1));
            CALC_2(23, 1, 0);

            _tmp = vtranslq_s8(vld1_s8(r2));
            CALC_1(23, 0);

            POSTPROCESS_2X4(_sum00, _sum10, outptr, outptr2, dstptr, dstptr2);

            r0 += 4;
            r1 += 4;
            r2 += 4;
            outptr += 4;
            outptr2 += 4;
            dstptr += 4;
            dstptr2 += 4;
        }
        r0 += tail_step + IW;
        r1 += tail_step + IW;
        r2 += tail_step + IW;
        outptr += OW;
        outptr2 += OW;
        dstptr += OW;
        dstptr2 += OW;
    }

    for (; h < OH; h++) {
        int w = 0;
        for (; w + 4 < width; w += 4) {
            int32x4x2_t _sum0, _sum1;
            if (!first_ic) {
                _sum0 = vld2q_s32(outptr);
                _sum1 = vld2q_s32(outptr + 8);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum0.val[0] = vdupq_n_s32(bptr[0]);
                    _sum0.val[1] = vdupq_n_s32(bptr[0]);
                    _sum1 = _sum0;
                } else {
                    _sum0.val[0] = vdupq_n_s32(0);
                    _sum1.val[0] = vdupq_n_s32(0);
                    _sum0.val[1] = vdupq_n_s32(0);
                    _sum1.val[1] = vdupq_n_s32(0);
                }
            }

            int8x16_t _r00 = vld1q_s8(r0);
            //! here will not not read out of bound
            int8x16_t _r01_ = vdupq_n_s8(r0[16]);
            int8x16_t _r10 = vld1q_s8(r1);
            int8x16_t _r11_ = vdupq_n_s8(r1[16]);
            int8x16_t _r01 = vextq_s8(_r00, _r01_, 1);
            int8x16_t _r11 = vextq_s8(_r10, _r11_, 1);

            int16x8x2_t r_20 =
                    vzipq_s16(vreinterpretq_s16_s8(_r00), vreinterpretq_s16_s8(_r10));
            int8x16_t _r0 = r_20.val[0];
            int8x16_t _r2 = r_20.val[1];

            int16x8x2_t r1_21 =
                    vzipq_s16(vreinterpretq_s16_s8(_r01), vreinterpretq_s16_s8(_r11));
            int8x16_t _r1 = r1_21.val[0];
            int8x16_t _r3 = r1_21.val[1];

            _sum0.val[0] = vdotq_s32(_sum0.val[0], _k, _r0);
            _sum0.val[1] = vdotq_s32(_sum0.val[1], _k, _r1);
            _sum1.val[0] = vdotq_s32(_sum1.val[0], _k, _r2);
            _sum1.val[1] = vdotq_s32(_sum1.val[1], _k, _r3);

            POSTPROCESS2_1X8(_sum0, outptr, dstptr);
            POSTPROCESS2_1X8(_sum1, outptr + 8, dstptr + 8);

            r0 += 16;
            r1 += 16;
            outptr += 16;
            dstptr += 16;
        }
        for (; w + 2 < width; w += 2) {
            int32x4_t _sum00, _sum01;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(1, 0);
            CALC_0(1, 1);

            _tmp = vld1q_s8(r1);
            CALC_0(23, 0);
            CALC_0(23, 1);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);

            r0 += 8;
            r1 += 8;
            outptr += 8;
            dstptr += 8;
        }

        for (; w < width; w++) {
            int32x4_t _sum00;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                } else {
                    _sum00 = vdupq_n_s32(0);
                }
            }

            _tmp = vtranslq_s8(vld1_s8(r0));
            CALC_0(1, 0);

            _tmp = vtranslq_s8(vld1_s8(r1));
            CALC_0(23, 0);

            POSTPROCESS_1X4(_sum00, outptr, dstptr);

            r0 += 4;
            r1 += 4;
            outptr += 4;
            dstptr += 4;
        }
        r0 += tail_step;
        r1 += tail_step;
    }
}

template <bool first_ic, bool last_ic, BiasMode bias_mode, typename Op>
MEGDNN_ATTRIBUTE_TARGET("dotprod")
void conv_bias::conv_direct_stride1_3x3_int8_dot(
        const int8_t* src, const int8_t* filter, const int32_t* bias, int32_t* temp,
        int8_t* dst, const size_t IH, const size_t IW, const size_t OH, const size_t OW,
        const Op& op) {
    MEGDNN_MARK_USED_VAR(IH);
    const size_t tail_step = IW - OW;

    const uint8x16_t _idx0 = {0, 1, 2, 16, 1, 2, 3, 16, 2, 3, 4, 16, 3, 4, 5, 16};
    const uint8x16_t _idx1 = {4, 5, 6, 16, 5, 6, 7, 16, 6, 7, 8, 16, 7, 8, 9, 16};
    const uint8x16_t _idx2 = {8,  9,  10, 16, 9,  10, 11, 16,
                              10, 11, 12, 16, 11, 12, 13, 16};
    int32_t* outptr = temp;
    int32_t* outptr2 = outptr + OW;
    int8_t* dstptr = dst;
    int8_t* dstptr2 = dstptr + OW;
    const int32_t* __restrict bptr = bias;

    const int8_t* r0 = src;
    const int8_t* r1 = src + IW;
    const int8_t* r2 = src + IW * 2;
    const int8_t* r3 = src + IW * 3;

    const int8_t* k0 = filter;

    int8x16_t _k_tmp = vcombine_s8(vld1_s8(k0), vdup_n_s8(k0[8]));
    uint8x16_t _idx = {0, 1, 2, 16, 0, 1, 2, 16, 0, 1, 2, 16, 0, 1, 2, 16};
    int8x16_t _k12 = vqtbl1q_s8_v7(_k_tmp, _idx);
    _idx = {3, 4, 5, 16, 3, 4, 5, 16, 3, 4, 5, 16, 3, 4, 5, 16};
    int8x16_t _k345 = vqtbl1q_s8_v7(_k_tmp, _idx);
    _idx = {6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16};
    int8x16_t _k678 = vqtbl1q_s8_v7(_k_tmp, _idx);

    int8x16_t _tmp, _elem;
    size_t h = 0;
    for (; h + 1 < OH; h += 2) {
        int width = OW >> 2;

        int w = 0;
        for (; w + 3 < width; w += 3) {
            int32x4_t _sum00, _sum01, _sum02, _sum10, _sum11, _sum12;

            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
                _sum02 = vld1q_s32(outptr + 8);
                _sum10 = vld1q_s32(outptr2);
                _sum11 = vld1q_s32(outptr2 + 4);
                _sum12 = vld1q_s32(outptr2 + 8);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                    _sum02 = _sum00;
                    _sum10 = _sum00;
                    _sum11 = _sum00;
                    _sum12 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                    _sum02 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                    _sum11 = vdupq_n_s32(0);
                    _sum12 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(12, 0);
            CALC_0(12, 1);
            CALC_0(12, 2);

            _tmp = vld1q_s8(r1);
            CALC_2(345, 12, 0);
            CALC_2(345, 12, 1);
            CALC_2(345, 12, 2);

            _tmp = vld1q_s8(r2);
            CALC_2(678, 345, 0);
            CALC_2(678, 345, 1);
            CALC_2(678, 345, 2);

            _tmp = vld1q_s8(r3);
            CALC_1(678, 0);
            CALC_1(678, 1);
            CALC_1(678, 2);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);
            POSTPROCESS_1X4(_sum02, outptr + 8, dstptr + 8);
            POSTPROCESS_1X8(_sum10, _sum11, outptr2, dstptr2);
            POSTPROCESS_1X4(_sum12, outptr2 + 8, dstptr2 + 8);

            r0 += 12;
            r1 += 12;
            r2 += 12;
            r3 += 12;
            outptr += 12;
            outptr2 += 12;
            dstptr += 12;
            dstptr2 += 12;
        }
        for (; w < width; w++) {
            int32x4_t _sum00, _sum10;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum10 = vld1q_s32(outptr2);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum10 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                }
            }

            _tmp = vtranslq_s8(vld1_s8(r0));
            CALC_0(12, 0);

            _tmp = vtranslq_s8(vld1_s8(r1));
            CALC_2(345, 12, 0);

            _tmp = vtranslq_s8(vld1_s8(r2));
            CALC_2(678, 345, 0);
            int8x8_t tmp_last = vreinterpret_s8_s32(vld1_dup_s32(r3));
            tmp_last = vreinterpret_s8_s16(
                    vld1_lane_s16(r3 + 4, vreinterpret_s16_s8(tmp_last), 2));
            _tmp = vtranslq_s8(tmp_last);
            CALC_1(678, 0);

            POSTPROCESS_2X4(_sum00, _sum10, outptr, outptr2, dstptr, dstptr2);

            r0 += 4;
            r1 += 4;
            r2 += 4;
            r3 += 4;
            outptr += 4;
            outptr2 += 4;
            dstptr += 4;
            dstptr2 += 4;
        }

        r0 += tail_step + IW;
        r1 += tail_step + IW;
        r2 += tail_step + IW;
        r3 += tail_step + IW;

        outptr += OW;
        outptr2 += OW;
        dstptr += OW;
        dstptr2 += OW;
    }

    for (; h < OH; h++) {
        int width = OW >> 2;

        int w = 0;
        for (; w + 3 < width; w += 3) {
            int32x4_t _sum00, _sum01, _sum02;

            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
                _sum02 = vld1q_s32(outptr + 8);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                    _sum02 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                    _sum02 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(12, 0);
            CALC_0(12, 1);
            CALC_0(12, 2);

            _tmp = vld1q_s8(r1);
            CALC_0(345, 0);
            CALC_0(345, 1);
            CALC_0(345, 2);

            _tmp = vld1q_s8(r2);
            CALC_0(678, 0);
            CALC_0(678, 1);
            CALC_0(678, 2);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);
            POSTPROCESS_1X4(_sum02, outptr + 8, dstptr + 8);

            r0 += 12;
            r1 += 12;
            r2 += 12;
            outptr += 12;
            dstptr += 12;
        }
        for (; w < width; w++) {
            int32x4_t _sum00;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                } else {
                    _sum00 = vdupq_n_s32(0);
                }
            }

            _tmp = vtranslq_s8(vld1_s8(r0));
            CALC_0(12, 0);

            _tmp = vtranslq_s8(vld1_s8(r1));
            CALC_0(345, 0);

            _tmp = vtranslq_s8(vld1_s8(r2));
            CALC_0(678, 0);

            POSTPROCESS_1X4(_sum00, outptr, dstptr);

            r0 += 4;
            r1 += 4;
            r2 += 4;
            outptr += 4;
            dstptr += 4;
        }
        r0 += tail_step;
        r1 += tail_step;
        r2 += tail_step;
    }
}

template <bool first_ic, bool last_ic, BiasMode bias_mode, typename Op>
MEGDNN_ATTRIBUTE_TARGET("dotprod")
void conv_bias::conv_direct_stride2_2x2_int8_dot(
        const int8_t* src, const int8_t* filter, const int32_t* bias, int32_t* temp,
        int8_t* dst, const size_t IH, const size_t IW, const size_t OH, const size_t OW,
        const Op& op) {
    MEGDNN_MARK_USED_VAR(IH);
    const size_t tail_step = IW - 2 * OW + IW;

    const uint8x16_t _idx0 = {0, 1, 16, 16, 2, 3, 16, 16, 4, 5, 16, 16, 6, 7, 16, 16};
    int32_t* outptr = temp;
    int8_t* dstptr = dst;

    const int8_t* r0 = src;
    const int8_t* r1 = src + IW;
    const int8_t* k0 = filter;
    const int32_t* __restrict bptr = bias;

    int8x16_t _k =
            vreinterpretq_s8_s32(vdupq_n_s32(*reinterpret_cast<const int32_t*>(k0)));
    uint8x16_t _idx = {0, 1, 16, 16, 0, 1, 16, 16, 0, 1, 16, 16, 0, 1, 16, 16};
    int8x16_t _k1 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {2, 3, 16, 16, 2, 3, 16, 16, 2, 3, 16, 16, 2, 3, 16, 16};
    int8x16_t _k23 = vqtbl1q_s8_v7(_k, _idx);

    int8x16_t _tmp, _elem;
    const int width = OW >> 2;
    size_t h = 0;
    for (; h < OH; h++) {
        int w = 0;
        for (; w + 2 < width; w += 2) {
            int32x4_t _sum0, _sum1;
            if (!first_ic) {
                _sum0 = vld1q_s32(outptr);
                _sum1 = vld1q_s32(outptr + 4);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum0 = vdupq_n_s32(bptr[0]);
                    _sum1 = _sum0;
                } else {
                    _sum0 = vdupq_n_s32(0);
                    _sum1 = vdupq_n_s32(0);
                }
            }

            int8x16_t _r00 = vld1q_s8(r0);
            //! here will not not read out of bound
            int8x16_t _r10 = vld1q_s8(r1);

            int16x8x2_t r_00 =
                    vzipq_s16(vreinterpretq_s16_s8(_r00), vreinterpretq_s16_s8(_r10));
            int8x16_t _r0 = r_00.val[0];
            int8x16_t _r1 = r_00.val[1];

            _sum0 = vdotq_s32(_sum0, _k, _r0);
            _sum1 = vdotq_s32(_sum1, _k, _r1);

            POSTPROCESS_1X8(_sum0, _sum1, outptr, dstptr);

            r0 += 16;
            r1 += 16;
            outptr += 8;
            dstptr += 8;
        }

        for (; w < width; w++) {
            int32x4_t _sum00;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                } else {
                    _sum00 = vdupq_n_s32(0);
                }
            }

            _tmp = vtranslq_s8(vld1_s8(r0));
            CALC_0(1, 0);

            _tmp = vtranslq_s8(vld1_s8(r1));
            CALC_0(23, 0);

            POSTPROCESS_1X4(_sum00, outptr, dstptr);

            r0 += 8;
            r1 += 8;
            outptr += 4;
            dstptr += 4;
        }
        r0 += tail_step;
        r1 += tail_step;
    }
}

template <bool first_ic, bool last_ic, BiasMode bias_mode, typename Op>
MEGDNN_ATTRIBUTE_TARGET("dotprod")
void conv_bias::conv_direct_stride2_3x3_int8_dot(
        const int8_t* src, const int8_t* filter, const int32_t* bias, int32_t* temp,
        int8_t* dst, const size_t IH, const size_t IW, const size_t OH, const size_t OW,
        const Op& op) {
    MEGDNN_MARK_USED_VAR(IH);
    const size_t tail_step = IW - 2 * OW + IW;

    const uint8x16_t _idx0 = {0, 1, 2, 16, 2, 3, 4, 16, 4, 5, 6, 16, 6, 7, 8, 16};
    int32_t* outptr = temp;
    int32_t* outptr2 = outptr + OW;
    int8_t* dstptr = dst;
    int8_t* dstptr2 = dstptr + OW;
    const int32_t* __restrict bptr = bias;

    const int8_t* r0 = src;
    const int8_t* r1 = src + IW;
    const int8_t* r2 = src + IW * 2;
    const int8_t* r3 = src + IW * 3;
    const int8_t* r4 = src + IW * 4;

    const int8_t* k0 = filter;

    int8x16_t _k_tmp = vcombine_s8(vld1_s8(k0), vdup_n_s8(k0[8]));
    uint8x16_t _idx = {0, 1, 2, 16, 0, 1, 2, 16, 0, 1, 2, 16, 0, 1, 2, 16};
    int8x16_t _k12 = vqtbl1q_s8_v7(_k_tmp, _idx);
    _idx = {3, 4, 5, 16, 3, 4, 5, 16, 3, 4, 5, 16, 3, 4, 5, 16};
    int8x16_t _k345 = vqtbl1q_s8_v7(_k_tmp, _idx);
    _idx = {6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16};
    int8x16_t _k678 = vqtbl1q_s8_v7(_k_tmp, _idx);

    int8x16_t _tmp, _elem;
    size_t h = 0;
    for (; h + 1 < OH; h += 2) {
        int width = OW >> 2;

        int w = 0;
        for (; w < width; w++) {
            int32x4_t _sum00, _sum10;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum10 = vld1q_s32(outptr2);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum10 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(12, 0);

            _tmp = vld1q_s8(r1);
            CALC_0(345, 0);

            _tmp = vld1q_s8(r2);
            CALC_2(678, 12, 0);

            _tmp = vld1q_s8(r3);
            CALC_1(345, 0);

            _tmp = vld1q_s8(r4);
            CALC_1(678, 0);

            POSTPROCESS_2X4(_sum00, _sum10, outptr, outptr2, dstptr, dstptr2);

            r0 += 8;
            r1 += 8;
            r2 += 8;
            r3 += 8;
            r4 += 8;
            outptr += 4;
            outptr2 += 4;
            dstptr += 4;
            dstptr2 += 4;
        }

        r0 += tail_step + IW * 2;
        r1 += tail_step + IW * 2;
        r2 += tail_step + IW * 2;
        r3 += tail_step + IW * 2;
        r4 += tail_step + IW * 2;

        outptr += OW;
        outptr2 += OW;
        dstptr += OW;
        dstptr2 += OW;
    }

    for (; h < OH; h++) {
        int width = OW >> 2;

        int w = 0;
        for (; w < width; w++) {
            int32x4_t _sum00;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                } else {
                    _sum00 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(12, 0);

            _tmp = vld1q_s8(r1);
            CALC_0(345, 0);

            _tmp = vld1q_s8(r2);
            CALC_0(678, 0);

            POSTPROCESS_1X4(_sum00, outptr, dstptr);

            r0 += 8;
            r1 += 8;
            r2 += 8;
            outptr += 4;
            dstptr += 4;
        }
        r0 += tail_step;
        r1 += tail_step;
        r2 += tail_step;
    }
}
#undef CALC_0
#undef CALC_1
#undef CALC_2

#define CALC_0(_k00_idx, _k01_idx, _c_idx)                         \
    _elem = vqtbl1q_s8_v7(_tmp, _idx##_c_idx##0);                  \
    _sum0##_c_idx = vdotq_s32(_sum0##_c_idx, _k##_k00_idx, _elem); \
    _elem = vqtbl1q_s8_v7(_tmp, _idx##_c_idx##1);                  \
    _sum0##_c_idx = vdotq_s32(_sum0##_c_idx, _k##_k01_idx, _elem);

#define CALC_1(_k00_idx, _k01_idx, _c_idx)                         \
    _elem = vqtbl1q_s8_v7(_tmp, _idx##_c_idx##0);                  \
    _sum1##_c_idx = vdotq_s32(_sum1##_c_idx, _k##_k00_idx, _elem); \
    _elem = vqtbl1q_s8_v7(_tmp, _idx##_c_idx##1);                  \
    _sum1##_c_idx = vdotq_s32(_sum1##_c_idx, _k##_k01_idx, _elem);

#define CALC_2(_k00_idx, _k01_idx, _k10_idx, _k11_idx, _c_idx)     \
    _elem = vqtbl1q_s8_v7(_tmp, _idx##_c_idx##0);                  \
    _sum0##_c_idx = vdotq_s32(_sum0##_c_idx, _k##_k00_idx, _elem); \
    _sum1##_c_idx = vdotq_s32(_sum1##_c_idx, _k##_k10_idx, _elem); \
    _elem = vqtbl1q_s8_v7(_tmp, _idx##_c_idx##1);                  \
    _sum0##_c_idx = vdotq_s32(_sum0##_c_idx, _k##_k01_idx, _elem); \
    _sum1##_c_idx = vdotq_s32(_sum1##_c_idx, _k##_k11_idx, _elem);

template <bool first_ic, bool last_ic, BiasMode bias_mode, typename Op>
MEGDNN_ATTRIBUTE_TARGET("dotprod")
void conv_bias::conv_direct_stride2_5x5_int8_dot(
        const int8_t* src, const int8_t* filter, const int32_t* bias, int32_t* temp,
        int8_t* dst, const size_t IH, const size_t IW, const size_t OH, const size_t OW,
        const Op& op) {
    MEGDNN_MARK_USED_VAR(IH);
    const size_t tail_step = IW - 2 * OW + IW;

    const uint8x16_t _idx00 = {0, 1, 2, 3, 2, 3, 4, 5, 4, 5, 6, 7, 6, 7, 8, 9};
    const uint8x16_t _idx01 = {4, 16, 16, 16, 6,  16, 16, 16,
                               8, 16, 16, 16, 10, 16, 16, 16};
    //! start from 8
    const uint8x16_t& _idx10 = _idx00;
    const uint8x16_t& _idx11 = _idx01;

    int8x16_t _tmp, _elem;
    int32_t* outptr = temp;
    int32_t* outptr2 = outptr + OW;
    int8_t* dstptr = dst;
    int8_t* dstptr2 = dstptr + OW;
    const int32_t* __restrict bptr = bias;

    const int8_t* r0 = src;
    const int8_t* r1 = src + IW;
    const int8_t* r2 = src + IW * 2;
    const int8_t* r3 = src + IW * 3;
    const int8_t* r4 = src + IW * 4;
    const int8_t* r5 = src + IW * 5;
    const int8_t* r6 = src + IW * 6;

    const int8_t* k0 = filter;

    int8x16_t _k = vld1q_s8(k0);
    //! filter row 1
    uint8x16_t _idx = {0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3};
    int8x16_t _k123 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {4, 16, 16, 16, 4, 16, 16, 16, 4, 16, 16, 16, 4, 16, 16, 16};
    int8x16_t _k4 = vqtbl1q_s8_v7(_k, _idx);
    //! filter row 2
    _idx = {5, 6, 7, 8, 5, 6, 7, 8, 5, 6, 7, 8, 5, 6, 7, 8};
    int8x16_t _k5678 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {9, 16, 16, 16, 9, 16, 16, 16, 9, 16, 16, 16, 9, 16, 16, 16};
    int8x16_t _k9 = vqtbl1q_s8_v7(_k, _idx);
    //! filter row 3
    _idx = {10, 11, 12, 13, 10, 11, 12, 13, 10, 11, 12, 13, 10, 11, 12, 13};
    int8x16_t _k10111213 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {14, 16, 16, 16, 14, 16, 16, 16, 14, 16, 16, 16, 14, 16, 16, 16};
    int8x16_t _k14 = vqtbl1q_s8_v7(_k, _idx);
    //! 9 10 11 12 -> 13 14 15 16 -> 17 18 19 20 -> 21 22 23 24
    _k = vld1q_s8(k0 + 9);
    //! filter row 4
    _idx = {6, 7, 8, 9, 6, 7, 8, 9, 6, 7, 8, 9, 6, 7, 8, 9};
    int8x16_t _k15161718 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {10, 16, 16, 16, 10, 16, 16, 16, 10, 16, 16, 16, 10, 16, 16, 16};
    int8x16_t _k19 = vqtbl1q_s8_v7(_k, _idx);
    //! filter row 5
    _idx = {11, 12, 13, 14, 11, 12, 13, 14, 11, 12, 13, 14, 11, 12, 13, 14};
    int8x16_t _k20212223 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {15, 16, 16, 16, 15, 16, 16, 16, 15, 16, 16, 16, 15, 16, 16, 16};
    int8x16_t _k24 = vqtbl1q_s8_v7(_k, _idx);

    const int width = OW >> 2;
    size_t h = 0;
    for (; h + 1 < OH; h += 2) {
        int w = 0;
        for (; w + 2 < width; w += 2) {
            int32x4_t _sum00, _sum01, _sum10, _sum11;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
                _sum10 = vld1q_s32(outptr2);
                _sum11 = vld1q_s32(outptr2 + 4);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                    _sum10 = _sum00;
                    _sum11 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                    _sum11 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 4, 0);
            _tmp = vld1q_s8(r0 + 8);
            CALC_0(123, 4, 1);

            _tmp = vld1q_s8(r1);
            CALC_0(5678, 9, 0);
            _tmp = vld1q_s8(r1 + 8);
            CALC_0(5678, 9, 1);

            _tmp = vld1q_s8(r2);
            CALC_2(10111213, 14, 123, 4, 0);
            _tmp = vld1q_s8(r2 + 8);
            CALC_2(10111213, 14, 123, 4, 1);

            _tmp = vld1q_s8(r3);
            CALC_2(15161718, 19, 5678, 9, 0);
            _tmp = vld1q_s8(r3 + 8);
            CALC_2(15161718, 19, 5678, 9, 1);

            _tmp = vld1q_s8(r4);
            CALC_2(20212223, 24, 10111213, 14, 0);
            _tmp = vld1q_s8(r4 + 8);
            CALC_2(20212223, 24, 10111213, 14, 1);

            _tmp = vld1q_s8(r5);
            CALC_1(15161718, 19, 0);
            _tmp = vld1q_s8(r5 + 8);
            CALC_1(15161718, 19, 1);

            _tmp = vld1q_s8(r6);
            CALC_1(20212223, 24, 0);
            _tmp = vld1q_s8(r6 + 8);
            CALC_1(20212223, 24, 1);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);
            POSTPROCESS_1X8(_sum10, _sum11, outptr2, dstptr2);

            r0 += 16;
            r1 += 16;
            r2 += 16;
            r3 += 16;
            r4 += 16;
            r5 += 16;
            r6 += 16;
            outptr += 8;
            outptr2 += 8;
            dstptr += 8;
            dstptr2 += 8;
        }
        for (; w < width; w++) {
            int32x4_t _sum00, _sum10;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum10 = vld1q_s32(outptr2);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum10 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 4, 0);

            _tmp = vld1q_s8(r1);
            CALC_0(5678, 9, 0);

            _tmp = vld1q_s8(r2);
            CALC_2(10111213, 14, 123, 4, 0);

            _tmp = vld1q_s8(r3);
            CALC_2(15161718, 19, 5678, 9, 0);

            _tmp = vld1q_s8(r4);
            CALC_2(20212223, 24, 10111213, 14, 0);

            _tmp = vld1q_s8(r5);
            CALC_1(15161718, 19, 0);

            _tmp = vld1q_s8(r6);
            CALC_1(20212223, 24, 0);

            POSTPROCESS_2X4(_sum00, _sum10, outptr, outptr2, dstptr, dstptr2);

            r0 += 8;
            r1 += 8;
            r2 += 8;
            r3 += 8;
            r4 += 8;
            r5 += 8;
            r6 += 8;
            outptr += 4;
            outptr2 += 4;
            dstptr += 4;
            dstptr2 += 4;
        }

        r0 += tail_step + IW * 2;
        r1 += tail_step + IW * 2;
        r2 += tail_step + IW * 2;
        r3 += tail_step + IW * 2;
        r4 += tail_step + IW * 2;
        r5 += tail_step + IW * 2;
        r6 += tail_step + IW * 2;

        outptr += OW;
        outptr2 += OW;
        dstptr += OW;
        dstptr2 += OW;
    }

    for (; h < OH; h++) {
        int w = 0;
        for (; w + 2 < width; w += 2) {
            int32x4_t _sum00, _sum01;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 4, 0);
            _tmp = vld1q_s8(r0 + 8);
            CALC_0(123, 4, 1);

            _tmp = vld1q_s8(r1);
            CALC_0(5678, 9, 0);
            _tmp = vld1q_s8(r1 + 8);
            CALC_0(5678, 9, 1);

            _tmp = vld1q_s8(r2);
            CALC_0(10111213, 14, 0);
            _tmp = vld1q_s8(r2 + 8);
            CALC_0(10111213, 14, 1);

            _tmp = vld1q_s8(r3);
            CALC_0(15161718, 19, 0);
            _tmp = vld1q_s8(r3 + 8);
            CALC_0(15161718, 19, 1);

            _tmp = vld1q_s8(r4);
            CALC_0(20212223, 24, 0);
            _tmp = vld1q_s8(r4 + 8);
            CALC_0(20212223, 24, 1);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);

            r0 += 16;
            r1 += 16;
            r2 += 16;
            r3 += 16;
            r4 += 16;
            outptr += 8;
            dstptr += 8;
        }
        for (; w < width; w++) {
            int32x4_t _sum00;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                } else {
                    _sum00 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 4, 0);

            _tmp = vld1q_s8(r1);
            CALC_0(5678, 9, 0);

            _tmp = vld1q_s8(r2);
            CALC_0(10111213, 14, 0);

            _tmp = vld1q_s8(r3);
            CALC_0(15161718, 19, 0);

            _tmp = vld1q_s8(r4);
            CALC_0(20212223, 24, 0);

            POSTPROCESS_1X4(_sum00, outptr, dstptr);

            r0 += 8;
            r1 += 8;
            r2 += 8;
            r3 += 8;
            r4 += 8;
            outptr += 4;
            dstptr += 4;
        }
        r0 += tail_step;
        r1 += tail_step;
        r2 += tail_step;
        r3 += tail_step;
        r4 += tail_step;
    }
}

template <bool first_ic, bool last_ic, BiasMode bias_mode, typename Op>
MEGDNN_ATTRIBUTE_TARGET("dotprod")
void conv_bias::conv_direct_stride2_7x7_int8_dot(
        const int8_t* src, const int8_t* filter, const int32_t* bias, int32_t* temp,
        int8_t* dst, const size_t IH, const size_t IW, const size_t OH, const size_t OW,
        const Op& op) {
    MEGDNN_MARK_USED_VAR(IH);
    const size_t tail_step = IW - 2 * OW + IW;

    const uint8x16_t _idx00 = {0, 1, 2, 3, 2, 3, 4, 5, 4, 5, 6, 7, 6, 7, 8, 9};
    const uint8x16_t _idx01 = {4, 5, 6, 16, 6, 7, 8, 16, 8, 9, 10, 16, 10, 11, 12, 16};
    //! start from 8
    const uint8x16_t& _idx10 = _idx00;
    const uint8x16_t& _idx11 = _idx01;

    int8x16_t _tmp, _elem;
    int32_t* outptr = temp;
    int32_t* outptr2 = outptr + OW;
    int8_t* dstptr = dst;
    int8_t* dstptr2 = dstptr + OW;
    const int32_t* __restrict bptr = bias;

    const int8_t* r0 = src;
    const int8_t* r1 = src + IW;
    const int8_t* r2 = src + IW * 2;
    const int8_t* r3 = src + IW * 3;
    const int8_t* r4 = src + IW * 4;
    const int8_t* r5 = src + IW * 5;
    const int8_t* r6 = src + IW * 6;
    const int8_t* r7 = src + IW * 7;
    const int8_t* r8 = src + IW * 8;

    const int8_t* k0 = filter;

    int8x16_t _k = vld1q_s8(k0);
    //! filter row 1
    uint8x16_t _idx = {0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3};
    int8x16_t _k123 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {4, 5, 6, 16, 4, 5, 6, 16, 4, 5, 6, 16, 4, 5, 6, 16};
    int8x16_t _k456 = vqtbl1q_s8_v7(_k, _idx);
    //! filter row 2
    _idx = {7, 8, 9, 10, 7, 8, 9, 10, 7, 8, 9, 10, 7, 8, 9, 10};
    int8x16_t _k78910 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {11, 12, 13, 16, 11, 12, 13, 16, 11, 12, 13, 16, 11, 12, 13, 16};
    int8x16_t _k111213 = vqtbl1q_s8_v7(_k, _idx);

    //! 12 13 14 15 -> 16 17 18 19 -> 20 21 22 23 -> 24 25 26 27
    _k = vld1q_s8(k0 + 12);
    //! filter row 3
    _idx = {2, 3, 4, 5, 2, 3, 4, 5, 2, 3, 4, 5, 2, 3, 4, 5};
    int8x16_t _k14151617 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16};
    int8x16_t _k181920 = vqtbl1q_s8_v7(_k, _idx);
    //! filter row 4
    _idx = {9, 10, 11, 12, 9, 10, 11, 12, 9, 10, 11, 12, 9, 10, 11, 12};
    int8x16_t _k21222324 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {13, 14, 15, 16, 13, 14, 15, 16, 13, 14, 15, 16, 13, 14, 15, 16};
    int8x16_t _k252627 = vqtbl1q_s8_v7(_k, _idx);

    //! 24 25 26 27->28 29 30 31 -> 32 33 34 35 -> 36 37 38 39
    _k = vld1q_s8(k0 + 24);
    //! filter row 5
    _idx = {4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7};
    int8x16_t _k28293031 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {8, 9, 10, 16, 8, 9, 10, 16, 8, 9, 10, 16, 8, 9, 10, 16};
    int8x16_t _k323334 = vqtbl1q_s8_v7(_k, _idx);

    //! 33 34 35 36 -> 37 38 39 40 -> 41 42 43 44 -> 45 46 47 48
    _k = vld1q_s8(k0 + 33);
    //! filter row 6
    _idx = {2, 3, 4, 5, 2, 3, 4, 5, 2, 3, 4, 5, 2, 3, 4, 5};
    int8x16_t _k35363738 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16};
    int8x16_t _k394041 = vqtbl1q_s8_v7(_k, _idx);

    //! filter row 7
    _idx = {9, 10, 11, 12, 9, 10, 11, 12, 9, 10, 11, 12, 9, 10, 11, 12};
    int8x16_t _k42434445 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {13, 14, 15, 16, 13, 14, 15, 16, 13, 14, 15, 16, 13, 14, 15, 16};
    int8x16_t _k464748 = vqtbl1q_s8_v7(_k, _idx);

    const int width = OW >> 2;
    size_t h = 0;
    for (; h + 1 < OH; h += 2) {
        int w = 0;
        for (; w + 2 < width; w += 2) {
            int32x4_t _sum00, _sum01, _sum10, _sum11;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
                _sum10 = vld1q_s32(outptr2);
                _sum11 = vld1q_s32(outptr2 + 4);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                    _sum10 = _sum00;
                    _sum11 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                    _sum11 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 456, 0);
            _tmp = vld1q_s8(r0 + 8);
            CALC_0(123, 456, 1);

            _tmp = vld1q_s8(r1);
            CALC_0(78910, 111213, 0);
            _tmp = vld1q_s8(r1 + 8);
            CALC_0(78910, 111213, 1);

            _tmp = vld1q_s8(r2);
            CALC_2(14151617, 181920, 123, 456, 0);
            _tmp = vld1q_s8(r2 + 8);
            CALC_2(14151617, 181920, 123, 456, 1);

            _tmp = vld1q_s8(r3);
            CALC_2(21222324, 252627, 78910, 111213, 0);
            _tmp = vld1q_s8(r3 + 8);
            CALC_2(21222324, 252627, 78910, 111213, 1);

            _tmp = vld1q_s8(r4);
            CALC_2(28293031, 323334, 14151617, 181920, 0);
            _tmp = vld1q_s8(r4 + 8);
            CALC_2(28293031, 323334, 14151617, 181920, 1);

            _tmp = vld1q_s8(r5);
            CALC_2(35363738, 394041, 21222324, 252627, 0);
            _tmp = vld1q_s8(r5 + 8);
            CALC_2(35363738, 394041, 21222324, 252627, 1);

            _tmp = vld1q_s8(r6);
            CALC_2(42434445, 464748, 28293031, 323334, 0);
            _tmp = vld1q_s8(r6 + 8);
            CALC_2(42434445, 464748, 28293031, 323334, 1);

            _tmp = vld1q_s8(r7);
            CALC_1(35363738, 394041, 0);
            _tmp = vld1q_s8(r7 + 8);
            CALC_1(35363738, 394041, 1);

            _tmp = vld1q_s8(r8);
            CALC_1(42434445, 464748, 0);
            _tmp = vld1q_s8(r8 + 8);
            CALC_1(42434445, 464748, 1);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);
            POSTPROCESS_1X8(_sum10, _sum11, outptr2, dstptr2);

            r0 += 16;
            r1 += 16;
            r2 += 16;
            r3 += 16;
            r4 += 16;
            r5 += 16;
            r6 += 16;
            r7 += 16;
            r8 += 16;
            outptr += 8;
            outptr2 += 8;
            dstptr += 8;
            dstptr2 += 8;
        }
        for (; w < width; w++) {
            int32x4_t _sum00, _sum10;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum10 = vld1q_s32(outptr2);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum10 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 456, 0);

            _tmp = vld1q_s8(r1);
            CALC_0(78910, 111213, 0);

            _tmp = vld1q_s8(r2);
            CALC_2(14151617, 181920, 123, 456, 0);

            _tmp = vld1q_s8(r3);
            CALC_2(21222324, 252627, 78910, 111213, 0);

            _tmp = vld1q_s8(r4);
            CALC_2(28293031, 323334, 14151617, 181920, 0);

            _tmp = vld1q_s8(r5);
            CALC_2(35363738, 394041, 21222324, 252627, 0);

            _tmp = vld1q_s8(r6);
            CALC_2(42434445, 464748, 28293031, 323334, 0);

            _tmp = vld1q_s8(r7);
            CALC_1(35363738, 394041, 0);

            _tmp = vld1q_s8(r8);
            CALC_1(42434445, 464748, 0);

            POSTPROCESS_2X4(_sum00, _sum10, outptr, outptr2, dstptr, dstptr2);

            r0 += 8;
            r1 += 8;
            r2 += 8;
            r3 += 8;
            r4 += 8;
            r5 += 8;
            r6 += 8;
            r7 += 8;
            r8 += 8;
            outptr += 4;
            outptr2 += 4;
            dstptr += 4;
            dstptr2 += 4;
        }

        r0 += tail_step + IW * 2;
        r1 += tail_step + IW * 2;
        r2 += tail_step + IW * 2;
        r3 += tail_step + IW * 2;
        r4 += tail_step + IW * 2;
        r5 += tail_step + IW * 2;
        r6 += tail_step + IW * 2;
        r7 += tail_step + IW * 2;
        r8 += tail_step + IW * 2;

        outptr += OW;
        outptr2 += OW;
        dstptr += OW;
        dstptr2 += OW;
    }

    for (; h < OH; h++) {
        int w = 0;
        for (; w + 2 < width; w += 2) {
            int32x4_t _sum00, _sum01;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 456, 0);
            _tmp = vld1q_s8(r0 + 8);
            CALC_0(123, 456, 1);

            _tmp = vld1q_s8(r1);
            CALC_0(78910, 111213, 0);
            _tmp = vld1q_s8(r1 + 8);
            CALC_0(78910, 111213, 1);

            _tmp = vld1q_s8(r2);
            CALC_0(14151617, 181920, 0);
            _tmp = vld1q_s8(r2 + 8);
            CALC_0(14151617, 181920, 1);

            _tmp = vld1q_s8(r3);
            CALC_0(21222324, 252627, 0);
            _tmp = vld1q_s8(r3 + 8);
            CALC_0(21222324, 252627, 1);

            _tmp = vld1q_s8(r4);
            CALC_0(28293031, 323334, 0);
            _tmp = vld1q_s8(r4 + 8);
            CALC_0(28293031, 323334, 1);

            _tmp = vld1q_s8(r5);
            CALC_0(35363738, 394041, 0);
            _tmp = vld1q_s8(r5 + 8);
            CALC_0(35363738, 394041, 1);

            _tmp = vld1q_s8(r6);
            CALC_0(42434445, 464748, 0);
            _tmp = vld1q_s8(r6 + 8);
            CALC_0(42434445, 464748, 1);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);

            r0 += 16;
            r1 += 16;
            r2 += 16;
            r3 += 16;
            r4 += 16;
            r5 += 16;
            r6 += 16;
            outptr += 8;
            dstptr += 8;
        }
        for (; w < width; w++) {
            int32x4_t _sum00;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                } else {
                    _sum00 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 456, 0);

            _tmp = vld1q_s8(r1);
            CALC_0(78910, 111213, 0);

            _tmp = vld1q_s8(r2);
            CALC_0(14151617, 181920, 0);

            _tmp = vld1q_s8(r3);
            CALC_0(21222324, 252627, 0);

            _tmp = vld1q_s8(r4);
            CALC_0(28293031, 323334, 0);

            _tmp = vld1q_s8(r5);
            CALC_0(35363738, 394041, 0);

            _tmp = vld1q_s8(r6);
            CALC_0(42434445, 464748, 0);

            POSTPROCESS_1X4(_sum00, outptr, dstptr);

            r0 += 8;
            r1 += 8;
            r2 += 8;
            r3 += 8;
            r4 += 8;
            r5 += 8;
            r6 += 8;
            outptr += 4;
            dstptr += 4;
        }
        r0 += tail_step;
        r1 += tail_step;
        r2 += tail_step;
        r3 += tail_step;
        r4 += tail_step;
        r5 += tail_step;
        r6 += tail_step;
    }
}

template <bool first_ic, bool last_ic, BiasMode bias_mode, typename Op>
MEGDNN_ATTRIBUTE_TARGET("dotprod")
void conv_bias::conv_direct_stride1_5x5_int8_dot(
        const int8_t* src, const int8_t* filter, const int32_t* bias, int32_t* temp,
        int8_t* dst, const size_t IH, const size_t IW, const size_t OH, const size_t OW,
        const Op& op) {
    MEGDNN_MARK_USED_VAR(IH);
    const size_t tail_step = IW - OW;

    const uint8x16_t _idx00 = {0, 1, 2, 3, 1, 2, 3, 4, 2, 3, 4, 5, 3, 4, 5, 6};
    const uint8x16_t _idx01 = {4, 16, 16, 16, 5, 16, 16, 16,
                               6, 16, 16, 16, 7, 16, 16, 16};
    const uint8x16_t _idx10 = {4, 5, 6, 7, 5, 6, 7, 8, 6, 7, 8, 9, 7, 8, 9, 10};
    const uint8x16_t _idx11 = {8,  16, 16, 16, 9,  16, 16, 16,
                               10, 16, 16, 16, 11, 16, 16, 16};
    const uint8x16_t _idx20 = {8,  9,  10, 11, 9,  10, 11, 12,
                               10, 11, 12, 13, 11, 12, 13, 14};
    const uint8x16_t _idx21 = {12, 16, 16, 16, 13, 16, 16, 16,
                               14, 16, 16, 16, 15, 16, 16, 16};
    int8x16_t _tmp, _elem;
    int32_t* outptr = temp;
    int32_t* outptr2 = outptr + OW;
    int8_t* dstptr = dst;
    int8_t* dstptr2 = dstptr + OW;
    const int32_t* __restrict bptr = bias;

    const int8_t* r0 = src;
    const int8_t* r1 = src + IW;
    const int8_t* r2 = src + IW * 2;
    const int8_t* r3 = src + IW * 3;
    const int8_t* r4 = src + IW * 4;
    const int8_t* r5 = src + IW * 5;

    const int8_t* k0 = filter;

    int8x16_t _k = vld1q_s8(k0);
    //! filter row 1
    uint8x16_t _idx = {0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3};
    int8x16_t _k123 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {4, 16, 16, 16, 4, 16, 16, 16, 4, 16, 16, 16, 4, 16, 16, 16};
    int8x16_t _k4 = vqtbl1q_s8_v7(_k, _idx);
    //! filter row 2
    _idx = {5, 6, 7, 8, 5, 6, 7, 8, 5, 6, 7, 8, 5, 6, 7, 8};
    int8x16_t _k5678 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {9, 16, 16, 16, 9, 16, 16, 16, 9, 16, 16, 16, 9, 16, 16, 16};
    int8x16_t _k9 = vqtbl1q_s8_v7(_k, _idx);
    //! filter row 3
    _idx = {10, 11, 12, 13, 10, 11, 12, 13, 10, 11, 12, 13, 10, 11, 12, 13};
    int8x16_t _k10111213 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {14, 16, 16, 16, 14, 16, 16, 16, 14, 16, 16, 16, 14, 16, 16, 16};
    int8x16_t _k14 = vqtbl1q_s8_v7(_k, _idx);
    //! 9 10 11 12 -> 13 14 15 16 -> 17 18 19 20 -> 21 22 23 24
    _k = vld1q_s8(k0 + 9);
    //! filter row 4
    _idx = {6, 7, 8, 9, 6, 7, 8, 9, 6, 7, 8, 9, 6, 7, 8, 9};
    int8x16_t _k15161718 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {10, 16, 16, 16, 10, 16, 16, 16, 10, 16, 16, 16, 10, 16, 16, 16};
    int8x16_t _k19 = vqtbl1q_s8_v7(_k, _idx);
    //! filter row 5
    _idx = {11, 12, 13, 14, 11, 12, 13, 14, 11, 12, 13, 14, 11, 12, 13, 14};
    int8x16_t _k20212223 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {15, 16, 16, 16, 15, 16, 16, 16, 15, 16, 16, 16, 15, 16, 16, 16};
    int8x16_t _k24 = vqtbl1q_s8_v7(_k, _idx);

    const int width = OW >> 2;
    size_t h = 0;
    for (; h + 1 < OH; h += 2) {
        int w = 0;
        for (; w + 3 < width; w += 3) {
            int32x4_t _sum00, _sum01, _sum02, _sum10, _sum11, _sum12;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
                _sum02 = vld1q_s32(outptr + 8);
                _sum10 = vld1q_s32(outptr2);
                _sum11 = vld1q_s32(outptr2 + 4);
                _sum12 = vld1q_s32(outptr2 + 8);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                    _sum02 = _sum00;
                    _sum10 = _sum00;
                    _sum11 = _sum00;
                    _sum12 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                    _sum02 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                    _sum11 = vdupq_n_s32(0);
                    _sum12 = vdupq_n_s32(0);
                }
            }
            _tmp = vld1q_s8(r0);
            CALC_0(123, 4, 0);
            CALC_0(123, 4, 1);
            CALC_0(123, 4, 2);

            _tmp = vld1q_s8(r1);
            CALC_2(5678, 9, 123, 4, 0);
            CALC_2(5678, 9, 123, 4, 1);
            CALC_2(5678, 9, 123, 4, 2);

            _tmp = vld1q_s8(r2);
            CALC_2(10111213, 14, 5678, 9, 0);
            CALC_2(10111213, 14, 5678, 9, 1);
            CALC_2(10111213, 14, 5678, 9, 2);

            _tmp = vld1q_s8(r3);
            CALC_2(15161718, 19, 10111213, 14, 0);
            CALC_2(15161718, 19, 10111213, 14, 1);
            CALC_2(15161718, 19, 10111213, 14, 2);

            _tmp = vld1q_s8(r4);
            CALC_2(20212223, 24, 15161718, 19, 0);
            CALC_2(20212223, 24, 15161718, 19, 1);
            CALC_2(20212223, 24, 15161718, 19, 2);

            _tmp = vld1q_s8(r5);
            CALC_1(20212223, 24, 0);
            CALC_1(20212223, 24, 1);
            CALC_1(20212223, 24, 2);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);
            POSTPROCESS_1X4(_sum02, outptr + 8, dstptr + 8);
            POSTPROCESS_1X8(_sum10, _sum11, outptr2, dstptr2);
            POSTPROCESS_1X4(_sum12, outptr2 + 8, dstptr2 + 8);

            r0 += 12;
            r1 += 12;
            r2 += 12;
            r3 += 12;
            r4 += 12;
            r5 += 12;
            outptr += 12;
            outptr2 += 12;
            dstptr += 12;
            dstptr2 += 12;
        }
        for (; w < width; w++) {
            int32x4_t _sum00, _sum10;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum10 = vld1q_s32(outptr2);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum10 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                }
            }

            _tmp = vtranslq_s8(vld1_s8(r0));
            CALC_0(123, 4, 0);

            _tmp = vtranslq_s8(vld1_s8(r1));
            CALC_2(5678, 9, 123, 4, 0);

            _tmp = vtranslq_s8(vld1_s8(r2));
            CALC_2(10111213, 14, 5678, 9, 0);

            _tmp = vtranslq_s8(vld1_s8(r3));
            CALC_2(15161718, 19, 10111213, 14, 0);

            _tmp = vtranslq_s8(vld1_s8(r4));
            CALC_2(20212223, 24, 15161718, 19, 0);

            _tmp = vtranslq_s8(vld1_s8(r5));
            CALC_1(20212223, 24, 0);

            POSTPROCESS_2X4(_sum00, _sum10, outptr, outptr2, dstptr, dstptr2);

            r0 += 4;
            r1 += 4;
            r2 += 4;
            r3 += 4;
            r4 += 4;
            r5 += 4;
            outptr += 4;
            outptr2 += 4;
            dstptr += 4;
            dstptr2 += 4;
        }

        r0 += tail_step + IW;
        r1 += tail_step + IW;
        r2 += tail_step + IW;
        r3 += tail_step + IW;
        r4 += tail_step + IW;
        r5 += tail_step + IW;

        outptr += OW;
        outptr2 += OW;
        dstptr += OW;
        dstptr2 += OW;
    }

    for (; h < OH; h++) {
        int w = 0;
        for (; w + 3 < width; w += 3) {
            int32x4_t _sum00, _sum01, _sum02;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
                _sum02 = vld1q_s32(outptr + 8);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                    _sum02 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                    _sum02 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 4, 0);
            CALC_0(123, 4, 1);
            CALC_0(123, 4, 2);

            _tmp = vld1q_s8(r1);
            CALC_0(5678, 9, 0);
            CALC_0(5678, 9, 1);
            CALC_0(5678, 9, 2);

            _tmp = vld1q_s8(r2);
            CALC_0(10111213, 14, 0);
            CALC_0(10111213, 14, 1);
            CALC_0(10111213, 14, 2);

            _tmp = vld1q_s8(r3);
            CALC_0(15161718, 19, 0);
            CALC_0(15161718, 19, 1);
            CALC_0(15161718, 19, 2);

            _tmp = vld1q_s8(r4);
            CALC_0(20212223, 24, 0);
            CALC_0(20212223, 24, 1);
            CALC_0(20212223, 24, 2);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);
            POSTPROCESS_1X4(_sum02, outptr + 8, dstptr + 8);

            r0 += 12;
            r1 += 12;
            r2 += 12;
            r3 += 12;
            r4 += 12;
            outptr += 12;
            dstptr += 12;
        }
        for (; w < width; w++) {
            int32x4_t _sum00;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                } else {
                    _sum00 = vdupq_n_s32(0);
                }
            }

            _tmp = vtranslq_s8(vld1_s8(r0));
            CALC_0(123, 4, 0);

            _tmp = vtranslq_s8(vld1_s8(r1));
            CALC_0(5678, 9, 0);

            _tmp = vtranslq_s8(vld1_s8(r2));
            CALC_0(10111213, 14, 0);

            _tmp = vtranslq_s8(vld1_s8(r3));
            CALC_0(15161718, 19, 0);

            _tmp = vtranslq_s8(vld1_s8(r4));
            CALC_0(20212223, 24, 0);

            POSTPROCESS_1X4(_sum00, outptr, dstptr);

            r0 += 4;
            r1 += 4;
            r2 += 4;
            r3 += 4;
            r4 += 4;
            outptr += 4;
            dstptr += 4;
        }
        r0 += tail_step;
        r1 += tail_step;
        r2 += tail_step;
        r3 += tail_step;
        r4 += tail_step;
    }
}

template <bool first_ic, bool last_ic, BiasMode bias_mode, typename Op>
MEGDNN_ATTRIBUTE_TARGET("dotprod")
void conv_bias::conv_direct_stride1_7x7_int8_dot(
        const int8_t* src, const int8_t* filter, const int32_t* bias, int32_t* temp,
        int8_t* dst, const size_t IH, const size_t IW, const size_t OH, const size_t OW,
        const Op& op) {
    MEGDNN_MARK_USED_VAR(IH);
    const size_t tail_step = IW - OW;

    const uint8x16_t _idx00 = {0, 1, 2, 3, 1, 2, 3, 4, 2, 3, 4, 5, 3, 4, 5, 6};
    const uint8x16_t _idx01 = {4, 5, 6, 16, 5, 6, 7, 16, 6, 7, 8, 16, 7, 8, 9, 16};
    const uint8x16_t _idx10 = {4, 5, 6, 7, 5, 6, 7, 8, 6, 7, 8, 9, 7, 8, 9, 10};
    const uint8x16_t _idx11 = {8,  9,  10, 16, 9,  10, 11, 16,
                               10, 11, 12, 16, 11, 12, 13, 16};

    int8x16_t _tmp, _elem;
    int32_t* outptr = temp;
    int32_t* outptr2 = outptr + OW;
    int8_t* dstptr = dst;
    int8_t* dstptr2 = dstptr + OW;
    const int32_t* __restrict bptr = bias;

    const int8_t* r0 = src;
    const int8_t* r1 = src + IW;
    const int8_t* r2 = src + IW * 2;
    const int8_t* r3 = src + IW * 3;
    const int8_t* r4 = src + IW * 4;
    const int8_t* r5 = src + IW * 5;
    const int8_t* r6 = src + IW * 6;
    const int8_t* r7 = src + IW * 7;

    const int8_t* k0 = filter;

    int8x16_t _k = vld1q_s8(k0);
    //! filter row 1
    uint8x16_t _idx = {0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3};
    int8x16_t _k123 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {4, 5, 6, 16, 4, 5, 6, 16, 4, 5, 6, 16, 4, 5, 6, 16};
    int8x16_t _k456 = vqtbl1q_s8_v7(_k, _idx);
    //! filter row 2
    _idx = {7, 8, 9, 10, 7, 8, 9, 10, 7, 8, 9, 10, 7, 8, 9, 10};
    int8x16_t _k78910 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {11, 12, 13, 16, 11, 12, 13, 16, 11, 12, 13, 16, 11, 12, 13, 16};
    int8x16_t _k111213 = vqtbl1q_s8_v7(_k, _idx);

    //! 12 13 14 15 -> 16 17 18 19 -> 20 21 22 23 -> 24 25 26 27
    _k = vld1q_s8(k0 + 12);
    //! filter row 3
    _idx = {2, 3, 4, 5, 2, 3, 4, 5, 2, 3, 4, 5, 2, 3, 4, 5};
    int8x16_t _k14151617 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16};
    int8x16_t _k181920 = vqtbl1q_s8_v7(_k, _idx);
    //! filter row 4
    _idx = {9, 10, 11, 12, 9, 10, 11, 12, 9, 10, 11, 12, 9, 10, 11, 12};
    int8x16_t _k21222324 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {13, 14, 15, 16, 13, 14, 15, 16, 13, 14, 15, 16, 13, 14, 15, 16};
    int8x16_t _k252627 = vqtbl1q_s8_v7(_k, _idx);

    //! 24 25 26 27->28 29 30 31 -> 32 33 34 35 -> 36 37 38 39
    _k = vld1q_s8(k0 + 24);
    //! filter row 5
    _idx = {4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7};
    int8x16_t _k28293031 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {8, 9, 10, 16, 8, 9, 10, 16, 8, 9, 10, 16, 8, 9, 10, 16};
    int8x16_t _k323334 = vqtbl1q_s8_v7(_k, _idx);

    //! 33 34 35 36 -> 37 38 39 40 -> 41 42 43 44 -> 45 46 47 48
    _k = vld1q_s8(k0 + 33);
    //! filter row 6
    _idx = {2, 3, 4, 5, 2, 3, 4, 5, 2, 3, 4, 5, 2, 3, 4, 5};
    int8x16_t _k35363738 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16, 6, 7, 8, 16};
    int8x16_t _k394041 = vqtbl1q_s8_v7(_k, _idx);

    //! filter row 7
    _idx = {9, 10, 11, 12, 9, 10, 11, 12, 9, 10, 11, 12, 9, 10, 11, 12};
    int8x16_t _k42434445 = vqtbl1q_s8_v7(_k, _idx);
    _idx = {13, 14, 15, 16, 13, 14, 15, 16, 13, 14, 15, 16, 13, 14, 15, 16};
    int8x16_t _k464748 = vqtbl1q_s8_v7(_k, _idx);

    const int width = OW >> 2;
    size_t h = 0;
    for (; h + 1 < OH; h += 2) {
        int w = 0;
        for (; w + 2 < width; w += 2) {
            int32x4_t _sum00, _sum01, _sum10, _sum11;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
                _sum10 = vld1q_s32(outptr2);
                _sum11 = vld1q_s32(outptr2 + 4);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                    _sum10 = _sum00;
                    _sum11 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                    _sum11 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 456, 0);
            CALC_0(123, 456, 1);

            _tmp = vld1q_s8(r1);
            CALC_2(78910, 111213, 123, 456, 0);
            CALC_2(78910, 111213, 123, 456, 1);

            _tmp = vld1q_s8(r2);
            CALC_2(14151617, 181920, 78910, 111213, 0);
            CALC_2(14151617, 181920, 78910, 111213, 1);

            _tmp = vld1q_s8(r3);
            CALC_2(21222324, 252627, 14151617, 181920, 0);
            CALC_2(21222324, 252627, 14151617, 181920, 1);

            _tmp = vld1q_s8(r4);
            CALC_2(28293031, 323334, 21222324, 252627, 0);
            CALC_2(28293031, 323334, 21222324, 252627, 1);

            _tmp = vld1q_s8(r5);
            CALC_2(35363738, 394041, 28293031, 323334, 0);
            CALC_2(35363738, 394041, 28293031, 323334, 1);

            _tmp = vld1q_s8(r6);
            CALC_2(42434445, 464748, 35363738, 394041, 0);
            CALC_2(42434445, 464748, 35363738, 394041, 1);

            _tmp = vld1q_s8(r7);
            CALC_1(42434445, 464748, 0);
            CALC_1(42434445, 464748, 1);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);
            POSTPROCESS_1X8(_sum10, _sum11, outptr2, dstptr2);

            r0 += 8;
            r1 += 8;
            r2 += 8;
            r3 += 8;
            r4 += 8;
            r5 += 8;
            r6 += 8;
            r7 += 8;
            outptr += 8;
            outptr2 += 8;
            dstptr += 8;
            dstptr2 += 8;
        }
        for (; w < width; w++) {
            int32x4_t _sum00, _sum10;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum10 = vld1q_s32(outptr2);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum10 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum10 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 456, 0);

            _tmp = vld1q_s8(r1);
            CALC_2(78910, 111213, 123, 456, 0);

            _tmp = vld1q_s8(r2);
            CALC_2(14151617, 181920, 78910, 111213, 0);

            _tmp = vld1q_s8(r3);
            CALC_2(21222324, 252627, 14151617, 181920, 0);

            _tmp = vld1q_s8(r4);
            CALC_2(28293031, 323334, 21222324, 252627, 0);

            _tmp = vld1q_s8(r5);
            CALC_2(35363738, 394041, 28293031, 323334, 0);

            _tmp = vld1q_s8(r6);
            CALC_2(42434445, 464748, 35363738, 394041, 0);

            _tmp = vld1q_s8(r7);
            CALC_1(42434445, 464748, 0);

            POSTPROCESS_2X4(_sum00, _sum10, outptr, outptr2, dstptr, dstptr2);

            r0 += 4;
            r1 += 4;
            r2 += 4;
            r3 += 4;
            r4 += 4;
            r5 += 4;
            r6 += 4;
            r7 += 4;
            outptr += 4;
            outptr2 += 4;
            dstptr += 4;
            dstptr2 += 4;
        }

        r0 += tail_step + IW;
        r1 += tail_step + IW;
        r2 += tail_step + IW;
        r3 += tail_step + IW;
        r4 += tail_step + IW;
        r5 += tail_step + IW;
        r6 += tail_step + IW;
        r7 += tail_step + IW;

        outptr += OW;
        outptr2 += OW;
        dstptr += OW;
        dstptr2 += OW;
    }

    for (; h < OH; h++) {
        int w = 0;
        for (; w + 2 < width; w += 2) {
            int32x4_t _sum00, _sum01;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
                _sum01 = vld1q_s32(outptr + 4);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                    _sum01 = _sum00;
                } else {
                    _sum00 = vdupq_n_s32(0);
                    _sum01 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 456, 0);
            CALC_0(123, 456, 1);

            _tmp = vld1q_s8(r1);
            CALC_0(78910, 111213, 0);
            CALC_0(78910, 111213, 1);

            _tmp = vld1q_s8(r2);
            CALC_0(14151617, 181920, 0);
            CALC_0(14151617, 181920, 1);

            _tmp = vld1q_s8(r3);
            CALC_0(21222324, 252627, 0);
            CALC_0(21222324, 252627, 1);

            _tmp = vld1q_s8(r4);
            CALC_0(28293031, 323334, 0);
            CALC_0(28293031, 323334, 1);

            _tmp = vld1q_s8(r5);
            CALC_0(35363738, 394041, 0);
            CALC_0(35363738, 394041, 1);

            _tmp = vld1q_s8(r6);
            CALC_0(42434445, 464748, 0);
            CALC_0(42434445, 464748, 1);

            POSTPROCESS_1X8(_sum00, _sum01, outptr, dstptr);

            r0 += 8;
            r1 += 8;
            r2 += 8;
            r3 += 8;
            r4 += 8;
            r5 += 8;
            r6 += 8;
            outptr += 8;
            dstptr += 8;
        }
        for (; w < width; w++) {
            int32x4_t _sum00;
            if (!first_ic) {
                _sum00 = vld1q_s32(outptr);
            } else {
                if (bias_mode == BiasMode::BROADCAST_CHANNEL_BIAS) {
                    _sum00 = vdupq_n_s32(bptr[0]);
                } else {
                    _sum00 = vdupq_n_s32(0);
                }
            }

            _tmp = vld1q_s8(r0);
            CALC_0(123, 456, 0);

            _tmp = vld1q_s8(r1);
            CALC_0(78910, 111213, 0);

            _tmp = vld1q_s8(r2);
            CALC_0(14151617, 181920, 0);

            _tmp = vld1q_s8(r3);
            CALC_0(21222324, 252627, 0);

            _tmp = vld1q_s8(r4);
            CALC_0(28293031, 323334, 0);

            _tmp = vld1q_s8(r5);
            CALC_0(35363738, 394041, 0);

            _tmp = vld1q_s8(r6);
            CALC_0(42434445, 464748, 0);

            POSTPROCESS_1X4(_sum00, outptr, dstptr);

            r0 += 4;
            r1 += 4;
            r2 += 4;
            r3 += 4;
            r4 += 4;
            r5 += 4;
            r6 += 4;
            outptr += 4;
            dstptr += 4;
        }
        r0 += tail_step;
        r1 += tail_step;
        r2 += tail_step;
        r3 += tail_step;
        r4 += tail_step;
        r5 += tail_step;
        r6 += tail_step;
    }
}

#undef CALC_0
#undef CALC_1
#undef CALC_2

#undef POSTPROCESS_1X8
#undef POSTPROCESS2_1X8
#undef POSTPROCESS_2X4
#undef POSTPROCESS_1X4
#undef ST1_S32X4
#undef ST2_S32X4X2

#define INSTANTIATION(stride, i, first_ic, last_ic, bias, Op)                \
    template void conv_bias::conv_direct_##stride##_##i##x##i##_int8_dot<    \
            first_ic, last_ic, bias, Op>(                                    \
            const int8_t*, const int8_t*, const int32_t*, int32_t*, int8_t*, \
            const size_t, const size_t, const size_t, const size_t, const Op&);

#define FOR_OP(stride, i, first_ic, last_ic, bias)      \
    INSTANTIATION(                                      \
            stride, i, first_ic, last_ic, bias,         \
            TypeCvtOp<dt_qint32 MEGDNN_COMMA dt_qint8>) \
    INSTANTIATION(                                      \
            stride, i, first_ic, last_ic, bias,         \
            ReluOp<dt_qint32 MEGDNN_COMMA dt_qint8>)    \
    INSTANTIATION(                                      \
            stride, i, first_ic, last_ic, bias,         \
            HSwishOp<dt_qint32 MEGDNN_COMMA dt_qint8>)

#define FOR_BIAS(stride, i, first_ic, last_ic)              \
    FOR_OP(stride, i, first_ic, last_ic, BiasMode::NO_BIAS) \
    FOR_OP(stride, i, first_ic, last_ic, BiasMode::BROADCAST_CHANNEL_BIAS)

#define FOR_IC(stride, i)             \
    FOR_BIAS(stride, i, true, true)   \
    FOR_BIAS(stride, i, true, false)  \
    FOR_BIAS(stride, i, false, false) \
    FOR_BIAS(stride, i, false, true)

#define FOR_FILTER(stride) \
    FOR_IC(stride, 2)      \
    FOR_IC(stride, 3)      \
    FOR_IC(stride, 5)      \
    FOR_IC(stride, 7)

#define FOR_STRIDE      \
    FOR_FILTER(stride1) \
    FOR_FILTER(stride2)

FOR_STRIDE

#undef FOR_STRIDE
#undef FOR_FILTER
#undef FOR_IC
#undef FOR_BIAS
#undef FOR_NONLINEAR
#undef INSTANTIATION

#endif
// vim: syntax=cpp.doxygen