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

Method on_kern_end

src/plugin/impl/num_range_checker.cpp:45–53  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

43}
44
45void NumRangeChecker::on_kern_end(const cg::event::OprExecKernelEnd& event) {
46 for (VarNode* var : event.opr->output()) {
47 if (!var->contain_flag(VarNode::Flag::VOLATILE_CONTENT) &&
48 var->dtype().category() == DTypeCategory::FLOAT) {
49 event.env->dispatch_on_comp_node(
50 var->comp_node(), [this, var]() { on_var_computed(var); });
51 }
52 }
53}
54
55void NumRangeChecker::on_subgraph_associated(
56 const cg::event::SubgraphAssociated& event) {

Callers

nothing calls this directly

Calls 6

categoryMethod · 0.80
outputMethod · 0.45
contain_flagMethod · 0.45
dtypeMethod · 0.45
dispatch_on_comp_nodeMethod · 0.45
comp_nodeMethod · 0.45

Tested by

no test coverage detected