Method on_var_computed

src/plugin/impl/num_range_checker.cpp:62–88

}

void NumRangeChecker::on_var_computed(VarNode* var) {
    if (!var->dev_tensor_valid())
        return;

    auto&& checker = m_cn2dt2checker[var->comp_node()][var->dtype().enumv()];
    checker.init(var, m_range);
    if (!checker.check(var)) {
        HostTensorND hv;
        hv.copy_from(var->dev_tensor()).sync();
        std::string msg{mgb_ssprintf_log(
                "float value out of range: var: %s\n",
                cg::dump_var_info({var}).c_str())};
        switch (hv.dtype().enumv()) {
#define cb(_dt)                                                 \
    case DTypeTrait<_dt>::enumv:                                \
        msg += format_msg<DTypeTrait<_dt>::ctype>(hv, m_range); \
        break;
            MEGDNN_FOREACH_COMPUTING_DTYPE_FLOAT(cb)
            default:
                mgb_assert(0, "unexpected dtype");
        }

        mgb_throw_raw(
                cg::OperatorNodeExcExtraInfo::ExcMaker{var->owner_opr()}.make<Error>(
                        msg));
    }
}

template <typename ctype>
std::string NumRangeChecker::format_msg(const HostTensorND& hv, float range) {