Class AllMLIRKernel

compiler/lib/CodeGen/CodeGen.cpp:12–22

using namespace KernelGen;
namespace {
struct AllMLIRKernel {
    AllMLIRKernel() {
        inner_map[KernelPack::KernType::ElemwiseKernel] =
                std::make_shared<codegen::ElemwiseKernel>();
        inner_map[KernelPack::KernType::MatrixMulKernel] =
                std::make_shared<codegen::MatmulKernel>();
    }

    std::unordered_map<KernelPack::KernType, std::shared_ptr<codegen::AutoKernelFunc>>
            inner_map;
};
}  // namespace

codegen::AutoKernelFunc* codegen::GenCode(KernelPack::KernType kern_type) {