Method IsAvailable

compiler/lib/CodeGen/Elemwise.cpp:33–41

using namespace mlir;

bool ElemwiseKernel::IsAvailable(TContext* context) const {
    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;
    return nr_operands_ok && (mode_ok_unary || mode_ok_binary || mode_ok_other);
}

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