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

Method GetKernelSymbol

compiler/lib/KernelGen/BareMetal/Concat.cpp:14–21  ·  view source on GitHub ↗

kernel gen

Source from the content-addressed store, hash-verified

12}
13//! kernel gen
14std::string ConcatKernel::GetKernelSymbol(TContext* context) const {
15 std::stringstream ss;
16 auto dtype_size = Utils::get_dtype_size(context->getAttrOprand("operand:0").dtype);
17
18 ss << "kernel_concat_byte_" << dtype_size << "_axis_"
19 << context->getAttrInt("axis");
20 return ss.str();
21}
22
23std::string ConcatKernel::GetKernelBody(TContext* context) const {
24 std::stringstream writer;

Callers

nothing calls this directly

Calls 2

get_dtype_sizeFunction · 0.85
getAttrOprandMethod · 0.80

Tested by

no test coverage detected