Class AllGICommonKernel

compiler/lib/KernelGen/GeneralIntrinsic/KernelPack.cpp:24–103

using namespace GeneralIntrinsic;
namespace {
struct AllGICommonKernel {
    AllGICommonKernel() {
        inner_map[KernelPack::KernType::ElemwiseKernel] = {
                std::make_shared<GeneralIntrinsic::ElemwiseKernel>()};
        inner_map[KernelPack::KernType::MatrixMulKernel] = {
                std::make_shared<GeneralIntrinsic::Fp16GevmKernel>(),
                std::make_shared<GeneralIntrinsic::Fp16GemvKernel>(),
                std::make_shared<GeneralIntrinsic::Fp16MatMulM8N8K8>(),
                std::make_shared<GeneralIntrinsic::Fp32MatMulM4N8K4>(),
                std::make_shared<GeneralIntrinsic::Fp32GevmKernel>(),
                std::make_shared<GeneralIntrinsic::Fp32GemvKernel>(),
                std::make_shared<GeneralIntrinsic::Fp32GemvMk4Kernel>(),
                std::make_shared<GeneralIntrinsic::Fp32MatMulM4N12>(),
                std::make_shared<GeneralIntrinsic::Fp32MatMulM4N12K4>()};

        inner_map[KernelPack::KernType::InternelKernel] = {
                std::make_shared<GeneralIntrinsic::MatmulM4N8MK4Kernel>(),
                std::make_shared<GeneralIntrinsic::MatmulM4N12Kernel>(),
                std::make_shared<GeneralIntrinsic::MatmulM4N12MK4Kernel>()};
        inner_map[KernelPack::KernType::ConvKernel] = {
                std::make_shared<GeneralIntrinsic::Conv1x1Float16MK8>(),
                std::make_shared<GeneralIntrinsic::ChannelWiseFloat16Mk8>(),
                std::make_shared<GeneralIntrinsic::ConvFloat16NCHWNCHW88>(),
                std::make_shared<GeneralIntrinsic::WinogradFp16F63NCHW88>(),
                std::make_shared<GeneralIntrinsic::WinogradFp16F43NCHW88>(),
                std::make_shared<GeneralIntrinsic::WinogradFp16F23NCHW88>(),
                std::make_shared<GeneralIntrinsic::ConvIm2colFloat16M8N8>(),
                std::make_shared<GeneralIntrinsic::Conv1x1FloatMk4>(),
                std::make_shared<GeneralIntrinsic::ChannelWiseFloatMk4>(),
                std::make_shared<GeneralIntrinsic::ConvFloatNCHWNCHW44>(),
                std::make_shared<GeneralIntrinsic::WinogradFloatF63NCHW44>(),
                std::make_shared<GeneralIntrinsic::WinogradFloatF43NCHW44>(),
                std::make_shared<GeneralIntrinsic::WinogradFloatF23NCHW44>(),
                std::make_shared<GeneralIntrinsic::ConvIm2colFloat>(),
                std::make_shared<GeneralIntrinsic::ConvIm2colFloatM4N8>(),

        };

        inner_map[KernelPack::KernType::ConvBackDataKernel] = {
                std::make_shared<GeneralIntrinsic::Float32NchwBackward>()};

        inner_map[KernelPack::KernType::PoolingKernel] = {
                std::make_shared<GeneralIntrinsic::PoolingNchw88Fp16>(),
                std::make_shared<GeneralIntrinsic::PoolingNchw44Fp32>(),
                std::make_shared<GeneralIntrinsic::PoolingNchw44QInt8>()};

        inner_map[KernelPack::KernType::TypeCvtKernel] = {
                std::make_shared<GeneralIntrinsic::TypecvtKernel>()};

        inner_map[KernelPack::KernType::ReduceKernel] = {
                std::make_shared<GeneralIntrinsic::ReduceKernel>()};

        inner_map[KernelPack::KernType::FlipKernel] = {
                std::make_shared<GeneralIntrinsic::FlipKernel>()};

        inner_map[KernelPack::KernType::ResizeKernel] = {
                std::make_shared<GeneralIntrinsic::ResizeKernel>()};