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

Method on_fwd_begin

src/opr/impl/loop/impl.cpp:1297–1315  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1295}
1296
1297void LoopImpl::MutableStateSaver::on_fwd_begin() {
1298 if (!m_enabled)
1299 return;
1300
1301 int swap_interval = m_swap_interval_setting;
1302 if (m_owner_opr->m_static_loop_time_infer) {
1303 int infer = m_owner_opr->m_static_loop_time_infer();
1304 if (swap_interval < 0)
1305 swap_interval = infer;
1306 else
1307 swap_interval = std::min(swap_interval, infer);
1308 } else {
1309 if (swap_interval < 0)
1310 swap_interval = -swap_interval;
1311 }
1312 mgb_assert(swap_interval > 0);
1313 for (auto&& i : m_var2info)
1314 i.second.recorder->setup_for_record(swap_interval);
1315}
1316
1317void LoopImpl::MutableStateSaver::on_fwd_finish() {
1318 if (!m_enabled)

Callers 1

scn_do_executeMethod · 0.80

Calls 2

setup_for_recordMethod · 0.80
minFunction · 0.50

Tested by

no test coverage detected