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

Method add_input_layout_constraint

src/opr-mm/impl/lock.cpp:57–72  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

55}
56
57void LockBase::add_input_layout_constraint() {
58 auto rst = owner_graph()
59 ->current_comp_seq()
60 ->user_data()
61 .get_user_data_or_create<LockGroupSet>();
62 {
63 MGB_LOCK_GUARD(rst->mtx);
64 m_cur_group = &rst->id2grp[m_param.group_id];
65 }
66 if (m_action == Action::ACQUIRE)
67 ++m_cur_group->nr_acquire;
68 else {
69 mgb_assert(m_action == Action::RELEASE);
70 ++m_cur_group->nr_release;
71 }
72}
73
74void LockBase::scn_do_execute_finish(const DeviceTensorND&) {
75 std::mutex* lock;

Callers

nothing calls this directly

Calls 2

user_dataMethod · 0.45
current_comp_seqMethod · 0.45

Tested by

no test coverage detected