Method GetCVKernelBody

compiler/lib/KernelGen/Arm/ArmCommon/CvtColor.cpp:453–482

}

std::string CvtColorKernel::GetCVKernelBody(TContext* context) const {
    auto kernel_sig = GetCVKernelSignature(context);
    auto mode = context->getAttrStr("mode");
    auto gen_func = g_cvt_reg.get_func(mode);
    auto code = gen_func(context);
    std::string body_temp = R"(
        #include <arm_neon.h>
        #include <limits.h>
        #include <string.h>
        #include "tinycv_c.h"
        ${aux_func}
        void ${kernel_sig}{
            uint8_t * sptr = src->data;
            uint8_t * dptr = dst->data;
            size_t rows = src->rows;
            size_t cols = src->cols;
            size_t hw = rows * cols;
            size_t ch = src->channels;
            
            ${cvt_code}
            
        }
    )";

    return StringTemplate::StringTemplateArgs()
            .add("aux_func", code.aux_func)
            .add("cvt_code", code.cvt_code)
            .add("kernel_sig", kernel_sig)
            .render(body_temp);
}

// vim: syntax=cpp.doxygen