Method on_var_computed

src/plugin/impl/var_value_checker.cpp:100–131

}

void VarValueChecker::on_var_computed(VarNode* var) {
    if (!var->dev_tensor_valid()) {
        if (m_init_val_dumped && var == m_vars[m_cur_var_idx]) {
            // skip vars that are not on device
            on_comp_seq_exec_finished({});
        }
        return;
    }

    if (!m_init_val_dumped) {
#if !__DEPLOY_ON_XP_SP2__
        m_var2val_mtx.lock();
#endif
        auto&& val = m_var2val[var];
#if !__DEPLOY_ON_XP_SP2__
        m_var2val_mtx.unlock();
#endif

        mgb_assert(!val);
        val = std::make_shared<DeviceTensorND>();
        val->copy_from(var->dev_tensor());
        return;
    }

    if (var != m_vars[m_cur_var_idx])
        return;

    if (!m_checker.valid()) {
        m_checker.init(var, m_var2val.at(var));
    }
    m_checker.check(var);
}

// vim: syntax=cpp.doxygen foldmethod=marker foldmarker=f{{{,f}}}