Method init

src/plugin/impl/num_range_checker.cpp:12–24

using namespace mgb;

void NumRangeChecker::Checker::init(VarNode* var, float range) {
    if (m_func)
        return;

    m_inp = std::make_shared<DeviceTensorND>(var->comp_node(), var->dtype());
    m_out = std::make_unique<HostTensorND>();
    auto cg = ComputingGraph::make();
    cg->options().log_level = 0;
    auto vi = opr::VolatileSharedDeviceTensor::make(*cg, m_inp),
         chk = opr::abs(vi) < range, good = opr::reduce_min(chk, chk.make_scalar(1));
    auto cb = [d = m_out.get()](DeviceTensorND& dv) { d->copy_from(dv).sync(); };
    m_func = cg->compile({{good, cb}});
}

bool NumRangeChecker::Checker::check(VarNode* var) {
    auto&& val = var->dev_tensor();