Method GetCVKernelBody

compiler/lib/KernelGen/GeneralIntrinsic/Resize.cpp:241–678

}

std::string ResizeKernel::GetCVKernelBody(TContext* context) const {
    auto kernel_sig = GetCVKernelSignature(context);
    auto src_dtype =
            Utils::cvt_dtype_specifier(context->getAttrOprand("operand:0").dtype);
    std::stringstream writer;
    writer << R"(
        #include <string.h>
        #include "tinycv_c.h"
        )";
    std::string body_temp = R"(
        #include <math.h>
        #include <stdalign.h>
        #include "${gi_header}"
        #define rep(i, n) for (int i = 0; i < (n); ++i)
        ${define_scale}
        ${output_converter_func}
        
        static inline void get_nearest_linear_coord(float scale, int size, int idx, float* ah0, int* ih0, float* ah1, int* ih1){
            if (size == 1) {
                *ah0 = 1.f;
                *ih0 = 0;
                *ah1 = 0.f;
                *ih1 = 0;
            }

            float alpha = (idx + 0.5f) / scale - 0.5f;
            int origin_idx = (int)(floorf(alpha));
            alpha -= origin_idx;

            if (origin_idx < 0) {
                origin_idx = 0;
                alpha = 0;
            } else if (origin_idx + 1 >= size) {
                origin_idx = size - 2;
                alpha = 1;
            }

            *ah0 = 1 - alpha;
            *ih0 = origin_idx;
            *ah1 = alpha;
            *ih1 = origin_idx + 1;
        }

        static inline void calc_cache_${func_tag}c3_1(const ${src_dtype}* src, const int src_h_stride, const ${src_dtype}* dst, const int OW,
                       const int* tabsx,
                       const int* tabsy,
                       const ${calc_dtype}* tabrx,
                       const ${calc_dtype}* tabry, int dx,
                       ${calc_dtype}* cache0, ${calc_dtype}* cache1) {
            (void)tabrx;
            const ${src_dtype}* psrc1 = src + (tabsx[dx] + 1) * src_h_stride;

            int dy = 0, dy3 = 0;

            for (; dy < OW; ++dy, dy3 += 3) {
                const ${src_dtype}* pcsrc10 = psrc1 + (tabsy[dy] + 0) * 3;
                const ${src_dtype}* pcsrc11 = psrc1 + (tabsy[dy] + 1) * 3;
                ${calc_dtype} ry = tabry[dy];