Function heuristic_tree

src/runtime/CL/mlgo/MLGOParser.cpp:709–776

}

void heuristic_tree(TokenStream &in, MLGOHeuristics &h, bool &valid)
{
    CHECK(expect_text(in, "<heuristic", valid), valid);
    const auto tree_id = CHECK(uint_val(in, valid), valid);
    CHECK(expect_text(in, ">", valid), valid);
    HeuristicTree *t                     = nullptr;
    std::tie(valid, t)                   = CHECK(h.get_heuristic_tree(tree_id), valid);
    const HeuristicType t_heuristic_type = std::get<0>(t->index());
    while (!accept_text(in, "</heuristic>"))
    {
        LOG_TOKEN_POS(in, pos);
        if (accept_text(in, "b"))
        {
            // Branch node
            const auto id   = CHECK(uint_val(in, valid), valid);
            const auto cond = CHECK(condition(in, valid), valid);
            const auto t_id = CHECK(uint_val(in, valid), valid);
            const auto f_id = CHECK(uint_val(in, valid), valid);
            valid           = CHECK(t->add_branch(id, cond, t_id, f_id), valid);
        }
        else if (accept_text(in, "l"))
        {
            // Leaf node
            const auto id = CHECK(uint_val(in, valid), valid);
            // NOTE: Heuristic type within each tree appears to be redundant (same information can be obtained from the
            // heuristic table). For now it remains as a step for validation.
            LOG_TOKEN_POS(in, pos);
            CHECK(expect_heuristic_type(in, t_heuristic_type, valid), valid);
            switch (t_heuristic_type)
            {
                case HeuristicType::GEMM_Type:
                {
                    const auto g_type = CHECK(gemm_type(in, valid), valid);
                    valid             = CHECK(t->add_leaf(id, g_type), valid);
                    break;
                }
                case HeuristicType::GEMM_Config_Native:
                {
                    const auto g_c = CHECK(gemm_config_native(in, valid), valid);
                    valid          = CHECK(t->add_leaf(id, g_c), valid);
                    break;
                }
                case HeuristicType::GEMM_Config_Reshaped_Only_RHS:
                {
                    const auto g_c = CHECK(gemm_config_reshaped_only_rhs(in, valid), valid);
                    valid          = CHECK(t->add_leaf(id, g_c), valid);
                    break;
                }
                case HeuristicType::GEMM_Config_Reshaped:
                {
                    const auto g_c = CHECK(gemm_config_reshaped(in, valid), valid);
                    valid          = CHECK(t->add_leaf(id, g_c), valid);
                    break;
                }
                default:
                {
                    FAIL_WITH_MSG(valid, pos, "Unexpected heuristic type");
                }