Method OnSetNodeDumpAttr

driver/cpu/cpu_driver.cpp:472–519

}

bool CPUDriver::OnSetNodeDumpAttr(DevContext* context, Subgraph* graph, const char* name, const void* buf, int size)
{
    if(size != sizeof(NodeDumpMsg))
    {
        set_tengine_errno(EINVAL);
        return false;
    }

    const NodeDumpMsg* msg = ( const NodeDumpMsg* )buf;
    const char* node_name = msg->node_name;

    Node* node = graph->FindNode(node_name);

    if(node == nullptr)
    {
        set_tengine_errno(EINVAL);
        return false;
    }

    if(!node->ExistAttr(ATTR_NODE_OPS))
    {
        set_tengine_errno(ENOTSUP);
        return false;
    }

    NodeOps* node_ops = any_cast<NodeOps*>(node->GetAttr(ATTR_NODE_OPS));

    int action = msg->action;

    switch(action)
    {
        case NODE_DUMP_ACTION_DISABLE:
            return node_ops->DisableDump(node);
            break;
        case NODE_DUMP_ACTION_ENABLE:
            return node_ops->EnableDump(node);
            break;
        case NODE_DUMP_ACTION_START:
            return node_ops->StartDump(node);
            break;
        case NODE_DUMP_ACTION_STOP:
            return node_ops->StopDump(node);
            break;
        default:
            set_tengine_errno(EINVAL);
            return false;
    }
}

bool CPUDriver::OnSetGraphPerfAttr(DevContext* context, Subgraph* graph, const char* name, const void* buf, int size)
{