Function BindOps

executor/lib/custom_kernel.cpp:156–195

}

NodeOps* BindOps(Node* node, const std::string& dev_name)
{
    CustomKernelList* k_list = any_cast<CustomKernelList>(&node->GetAttr(ATTR_CUSTOM_KERNEL));

    struct custom_kernel_ops* ops = k_list->GetKernel(dev_name.c_str());

    if(ops == nullptr)
    {
        /* try if there is any device support setting */
        ops = k_list->GetKernel(ANY_DEVICE_NAME);

        if(ops == nullptr)
            return nullptr;
    }

    Operator* op = node->GetOp();

    if(op->GetName() != "Generic" && (op->GetName() != ops->op))
    {
        XLOG_WARN() << "unmatched op: custom " << ops->op << " real " << op->GetName() << "\n";
        return nullptr;
    }

    /* check bind */
    if(ops->bind != nullptr && !CheckBind(node, ops))
    {
        if(ops->force)
        {
            LOG_ERROR() << "bind custom kernel for node: " << node->GetName() << " failed, but force is set\n";

            set_tengine_errno(ENOTRECOVERABLE);
        }

        return nullptr;
    }

    NodeOps* node_ops = CustomKernelNodeOps::NewOps(node, ops);

    return node_ops;
}

}    // namespace CustomKernelManager