MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / opr_register

Method opr_register

src/opr-mm/impl/mm_handler.cpp:73–83  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

71 req.ParseFromArray(input_ptr, input_len);
72
73void GroupServerProxy::opr_register(
74 void* input_ptr, size_t input_len, std::string* output) {
75 INFO_INIT(mm_handler, OprRegister);
76 auto ret = m_mgr.opr_register(
77 req.key(), req.nr_expected_devices(), req.is_root(), req.rank(),
78 req.comp_node_hash());
79 rsp.set_hash(ret.hash);
80 rsp.set_rank(ret.rank);
81 rsp.set_root_rank(ret.root_rank);
82 rsp.SerializeToString(output);
83}
84
85void GroupServerProxy::set_output_shape(
86 void* input_ptr, size_t input_len, std::string* output) {

Callers 1

scn_do_executeMethod · 0.45

Calls 4

keyMethod · 0.45
is_rootMethod · 0.45
rankMethod · 0.45
hashMethod · 0.45

Tested by

no test coverage detected