Method CondExecPred

src/opr/impl/cond.cpp:316–353

}

CondExecPred::CondExecPred(
        VarNode* pred, const VarNodeArrayView& keys, const Param& param,
        const OperatorNodeConfig& config)
        : Super(pred->owner_graph(), config, "cond_pred", {pred}), m_param{param} {
    m_masks.reserve(keys.size() + 1);
    auto add_out = [this](const std::string& name) {
        auto var = add_output(name);
        var->add_flag(VarNode::Flag::NO_SYS_MEM_ALLOC).dtype(dtype::Int32{});
        m_masks.emplace_back(std::make_shared<ExecutionMask>(var));
    };
    for (size_t i = 0; i < keys.size(); ++i) {
        mgb_throw_if(
                keys[i]->dtype() != pred->dtype(), GraphError,
                "dtype mismatch: pred=%s input[%zu]=%s", pred->dtype().name(), i,
                keys[i]->dtype().name());
        add_input({keys[i]});
        if (param.mode == Param::Mode::PIECEWISE) {
            if (!i) {
                add_out("[-inf,k0]");
            }
            if (i != keys.size() - 1) {
                add_out(ssprintf("[k%zu,k%zu]", i, i + 1));
            } else {
                add_out(ssprintf("[k%zu,inf]", i));
            }
        } else {
            add_out(ssprintf("branch%zu", i));
        }
    }
    if (param.mode == Param::Mode::CASE_FALLBACK) {
        add_out("fallback");
    }
    add_input({pred});
    add_equivalence_component<PODHash<Param>>(&m_param);

    // ensure listener is registered
    GlobalRegistry::get(*owner_graph());
}

cg::OperatorNodeBase* CondExecPred::make_opr(
        SymbolVar pred, const VarNodeArrayView& keys, const Param& param,