Method IsAvailable

compiler/lib/KernelGen/AutoBareMetal/ElemwiseKernel.cpp:14–25

using namespace AutoBareMetal;

bool ElmwiseKernel::IsAvailable(TContext* context) const {
    auto auto_kern = codegen::GenCode(KernelPack::KernType::ElemwiseKernel);
    auto mode = context->getAttrStr("mode");
    auto nr_operands = context->getAttrInt("nr_operands");
    bool nr_operands_ok = nr_operands == 2;
    bool mode_ok_unary = mode == "RELU";
    bool mode_ok_binary = false;
    bool mode_ok_other = false;
    bool mode_kern_ok = auto_kern->IsAvailable(context);
    return nr_operands_ok && (mode_ok_unary || mode_ok_binary || mode_ok_other) &&
           mode_kern_ok;
}

std::string ElmwiseKernel::GetKernelSymbol(TContext* context) const {
    std::stringstream ss;