| 174 | } |
| 175 | |
| 176 | void execute_on_comp_node(const CompNode& comp_node) { |
| 177 | for (auto&& state : m_var_state) { |
| 178 | auto s0 = state.front().get(); |
| 179 | if (s0->var->comp_node() == comp_node) |
| 180 | s0->set_ready(); |
| 181 | } |
| 182 | for (auto&& instr : m_copy_instr.at(comp_node)) { |
| 183 | instr.execute(); |
| 184 | } |
| 185 | } |
| 186 | }; |
| 187 | |
| 188 | void AllGather::get_output_var_shape( |