Method GetCVKernelBody

compiler/lib/KernelGen/GeneralIntrinsic/CVTranspose.cpp:43–71

             common_tran.GetDependInternalSymbol(ctx)}};
}
std::string CvTransposeKernel::GetCVKernelBody(TContext* context) const {
    GeneralIntrinsic::CommonTransposeKernel common_tran;
    std::shared_ptr<TContext> tran_ctx = std::make_shared<CodeGenContext>();
    auto kernel_sig = GetCVKernelSignature(context);
    auto dtype_str = context->getAttrOprand("operand:0").dtype;
    int32_t type_size = Utils::get_dtype_size(dtype_str);
    tran_ctx->setAttr("type_size", CCAttr(type_size));
    std::string body_temp = R"(
        #include <string.h>
        #include "gi_int.h"
        #include "tinycv_c.h"

        ${gi_fast_transpose_impl}

        void ${kernel_sig}{
            uint8_t * src_base_ptr = src->data;
            uint8_t * dst_base_ptr = dst->data;
            int src_cols = src->cols;
            int src_rows = src->rows;
            int src_chans = src->channels;
            fast_transpose_impl_8(src_base_ptr, dst_base_ptr, src_cols, src_rows, src_chans, src_cols * src_chans, src_rows * src_chans);
        }
    )";
    return StringTemplate::StringTemplateArgs()
            .add("kernel_sig", kernel_sig)
            .add("gi_fast_transpose_impl",
                 common_tran.GetKernelSignature(tran_ctx.get()))
            .render(body_temp);
}

// vim: syntax=cpp.doxygen