MCPcopy Create free account
hub / github.com/MegEngine/MegCC / GetCVKernelSubSymbol

Method GetCVKernelSubSymbol

compiler/lib/KernelGen/Arm/ArmCommon/CvtColor.cpp:440–447  ·  view source on GitHub ↗

kernel gen

Source from the content-addressed store, hash-verified

438
439//! kernel gen
440std::string CvtColorKernel::GetCVKernelSubSymbol(TContext* context) const {
441 std::stringstream ss;
442 auto src_dtype = context->getAttrOprand("operand:0").dtype;
443 auto mode = context->getAttrStr("mode");
444 auto mode_str = g_cvt_reg.get_mode_sym(mode);
445 ss << "tinycv_cvt_" << mode_str << "_" << src_dtype;
446 return ss.str();
447}
448
449std::string CvtColorKernel::GetCVKernelSignature(TContext* context) const {
450 return GetCVKernelSymbol(context) + "(const TinyMat* src, const TinyMat* dst)";

Callers

nothing calls this directly

Calls 3

getAttrOprandMethod · 0.80
getAttrStrMethod · 0.80
get_mode_symMethod · 0.45

Tested by

no test coverage detected