| 20 | #if MGB_ENABLE_OPR_MM |
| 21 | namespace { |
| 22 | cg::OperatorNodeBase* apply_on_var_node_remote_send( |
| 23 | const OpDef& def, const VarNodeArray& inputs) { |
| 24 | auto&& send = def.cast_final_safe<RemoteSend>(); |
| 25 | auto group_client = std::make_shared<opr::GroupClientProxy>( |
| 26 | ssprintf("%s:%d", send.addr.data(), send.port)); |
| 27 | auto&& graph = inputs[0]->owner_graph(); |
| 28 | |
| 29 | OperatorNodeConfig config{send.make_name()}; |
| 30 | cg::OperatorNodeBase* opr = |
| 31 | graph->insert_opr(std::make_unique<mgb::opr::RemoteSend>( |
| 32 | send.key, inputs[0], group_client, true, send.backend, config)); |
| 33 | return opr; |
| 34 | } |
| 35 | |
| 36 | cg::OperatorNodeBase* apply_on_var_node_remote_recv( |
| 37 | const OpDef& def, const VarNodeArray& inputs) { |
nothing calls this directly
no test coverage detected