Method do_execute

src/opr/impl/muxing.cpp:259–271

}

void AllGather::do_execute(ExecEnv& env) {
    CompNode::UnorderedSet used_cn;
    for (auto i : output()) {
        if (!used_cn.insert(i->comp_node()).second)
            continue;
        auto runner = [this, cn = i->comp_node()]() {
            owner_graph()->event().signal_inplace<cg::event::BeforeKernel>(this, cn);
            m_copy_strategy->execute_on_comp_node(cn);
            owner_graph()->event().signal_inplace<cg::event::AfterKernel>(this, cn);
        };
        env.dispatch_on_comp_node(i->comp_node(), runner);
    }
}

VarNodeArray AllGather::grad(const VarNodeArray& out_grad) {
    CompNode::UnorderedMap<VarNode*> cn_reduced;