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

Method current_exec_env

src/core/impl/graph/cg_impl.cpp:864–872  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

862}
863
864GraphExecutable::ExecEnv* ComputingGraphImpl::current_exec_env() {
865 if (auto ret = eager_eval_manager().exec_env()) {
866 return ret;
867 }
868 if (m_current_comp_seq) {
869 return &static_cast<ComputingSequence*>(m_current_comp_seq)->exec_env();
870 }
871 return nullptr;
872}
873
874Maybe<size_t> ComputingGraphImpl::opr_step_num_in_cur_comp_seq(OperatorNodeBase* opr) {
875 mgb_assert(m_current_comp_seq && opr->owner_graph() == this);

Callers 2

TESTFunction · 0.80
defragMethod · 0.80

Calls 1

exec_envMethod · 0.45

Tested by 1

TESTFunction · 0.64