Method execute

src/core/impl/graph/operator_node.cpp:87–165

}

void OperatorNodeBase::execute(ExecEnv& env) {
    if (owner_graph()->options().imperative_proxy_graph) {
        do_execute(env);
        return;
    }

    owner_graph()->event().signal_inplace<event::OprExecStart>(this, &env);

    // dispatch waiting commands
    for (auto&& wspec : input_waiting_spec()) {
        auto runner = [this, ps = &wspec]() {
            for (VarNode* i : ps->dev_ready) {
                auto&& event = VarNodeMemManager::var_node_cn_sync_manager(i)
                                       ->busy_wait_set_ready_and_get_event();
                ps->comp_node.device_wait_event(event);
            }
            owner_graph()->event().signal_inplace<event::AfterWait>(
                    ps->comp_node, this);
        };
        // always maintain var sync order, so we dispatch without execution mask
        env.dispatch_on_comp_node_with_mask(wspec.comp_node, runner, nullptr);
    }

    // allocate output with dynamic storage
    ComputingGraphImpl::downcast(owner_graph())
            ->var_node_mem_manager()
            .alloc_var_node_mem_dynamic(env, this);

    // find shape-dep inputs:
    // shape/value deps whose static infer source is missing are added to
    // DEV_COMP_ORDER dep by topo sorter, so it is guaranteed that static infer
    // would success here. For host-value deps, opr would query
    // static_infer_manager so the value would be up-to-date; however for shape
    // deps, oprs would access the shape directly, so we need to insert some
    // code here to ensure it is up-to-date.
    if (!ComputingGraphImpl::downcast(owner_graph())->eager_eval_manager().enabled()) {
        VarNodeArray vars_to_set;
        auto cg = ComputingGraphImpl::downcast(owner_graph());
        auto step_cur = cg->opr_step_num_in_cur_comp_seq(this).val();
        mgb_assert(step_cur < std::numeric_limits<size_t>::max());
        using DT = NodeProp::DepType;
        CompNode uniq_cn;  // all outputs should be on the same comp node
        for (auto&& i : node_prop().dep_map()) {
            if ((i.second & DT::SHAPE) && !(i.second & DT::DEV_VALUE)) {
                auto var = i.first;
                if (!uniq_cn.valid()) {
                    uniq_cn = output(0)->comp_node();
                    for (auto i : output()) {
                        mgb_assert(
                                uniq_cn == i->comp_node(),
                                "opr that has shape dep should be on a "
                                "single comp node; opr=%s{%s}",
                                cname(), dyn_typeinfo()->name);
                    }
                }

                auto vs = cg->opr_step_num_in_cur_comp_seq(var->owner_opr());
                if (!vs.valid() || step_cur < vs.val() || var->comp_node() != uniq_cn) {