Method RegisterOPImplementor

executor/lib/node_ops.cpp:327–354

}

bool NodeOpsRegistryManager::RegisterOPImplementor(const std::string& registry_name, const std::string& op_name,
                                                   select_node_ops_t select_func, int priority)
{
    // TODO: Add Lock to protect find and register registry

    auto registry = FindRegistry(registry_name);

    if(registry == nullptr)
    {
        registry = new NodeOpsRegistry(registry_name);
        NodeOpsRegistryManager::AddRegistry(registry->reg_name, registry);
    }

    PrioSelector* prio_selector = dynamic_cast<PrioSelector*>(registry->FindSelector(op_name));

    if(prio_selector == nullptr)
    {
        prio_selector = new PrioSelector();
        prio_selector->op_name = op_name;

        registry->RegisterSelector(prio_selector);
    }

    if(!prio_selector->Register(priority, select_func))
        return false;

    return true;
}

NodeOps* NodeOpsRegistry::FindNodeOps(const CPUInfo* cpu_info, Node* node)
{