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

Method on_comp_seq_finished

src/plugin/impl/infkern_finder.cpp:224–236  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

222}
223
224void InfkernFinder::on_comp_seq_finished(const cg::event::CompSeqExecFinished& ev) {
225 mgb_assert(ev.exec == m_current_comp_seq);
226 if (!ev.device_actually_finished)
227 return;
228
229 if (m_global_state_storage) {
230 mgb_log("InfkernFinder: computing graph %p finished", m_owner_graph);
231 m_cg_start_log_printed.clear();
232 }
233 for (auto&& i : m_opr_seq)
234 i.check_event_finished();
235 m_prev_succ_comp_seq_run_id = m_current_comp_seq->get_run_id();
236}
237
238void InfkernFinder::on_opr_start(const cg::event::OprExecStart& ev) {
239 m_opr2state.at(ev.opr)->on_opr_start(ev.env);

Callers

nothing calls this directly

Calls 3

check_event_finishedMethod · 0.80
clearMethod · 0.45
get_run_idMethod · 0.45

Tested by

no test coverage detected