GCC Plugin, добавьте новую оптимизирующую прагму

Question

Я создаю плагин GCC.

Я пытаюсь создать плагин для конкретного преобразования цикла - развернуть цикл ровно N (заданный параметр) раз. Я правильно установил плагины и могу успешно зарегистрировать свою прагму в процессе компиляции. Когда я регистрирую прагму с функцией c_register_pragmaЯ могу справиться с этим в лексическом анализе (с функцией handle_my_pragma), но как я могу найти это тогда?

Я также могу определить свой собственный проход и обход GIMPLE, но нет никаких следов прагмы. Итак, мой вопрос: где моя прагма и как я могу повлиять на мой код? Или что бы вы предложили для достижения моей цели? Это не должно быть с прагмой, но это казалось хорошей идеей. Также я знаю о MELT, но в рамках изучения GCC я бы предпочел чистый плагин на C.

Мой код

static bool looplugin_gate(void)
{
    return true;
}

static unsigned looplugin_exec(void)
{
    printf( "===looplugin_exec===\n" );

    basic_block bb;
    gimple stmt;
    gimple_stmt_iterator gsi;

    FOR_EACH_BB(bb)
    {
        for (gsi=gsi_start_bb(bb); !gsi_end_p(gsi); gsi_next(&gsi), j++)
        {
            stmt = gsi_stmt(gsi);
            print_gimple_stmt (stdout, stmt, 0, TDF_SLIM);
        }
    }
    return 0;
}


void handle_my_pragma(cpp_reader *ARG_UNUSED(dummy))
{
    printf ("=======Handling loopragma=======\n" );
    enum cpp_ttype token;
    tree x;
    int num = -1;

    token = pragma_lex (&x);
    if (TREE_CODE (x) != INTEGER_CST)
        warning (0, "invalid constant in %<#pragma looppragma%> - ignored"); 
    num = TREE_INT_CST_LOW (x);
    printf( "Detected #pragma loopragma %d\n", num );
}

static void register_my_pragma (void *event_data, void *data)
{
    warning (0, G_("Callback to register pragmas"));
    c_register_pragma (NULL, "loopragma", handle_my_pragma);
}


static struct opt_pass myopt_pass = 
{
    .type = GIMPLE_PASS,
    .name = "LoopPlugin",
    .gate = looplugin_gate,
    .execute = looplugin_exec
};

int plugin_init(struct plugin_name_args   *info,  /* Argument infor */
struct plugin_gcc_version *ver)   /* Version of GCC */
{
const char * plugin_name = info->base_name;
struct register_pass_info pass;

pass.pass = &myopt_pass;
pass.reference_pass_name = "ssa";
pass.ref_pass_instance_number = 1;
pass.pos_op = PASS_POS_INSERT_BEFORE;

register_callback( plugin_name, PLUGIN_PRAGMAS, register_my_pragma, NULL );
register_callback( plugin_name, PLUGIN_PASS_MANAGER_SETUP, NULL, &pass );

return 0;
}

PS: Если кто-то был знаком с разработкой плагинов GCC и имел доброе сердце:), пожалуйста, свяжитесь со мной (mbukovy gmail com). Я делаю это из-за моей последней диссертации (по собственному выбору), и я приветствую любую родственную душу.

Answer 1

Когда я регистрирую прагму с помощью функции c_register_pragma, я могу обработать ее в лексическом анализе (с помощью функции handle_my_pragma), но как я могу найти ее тогда?

Существует опция (на самом деле, хак) для создания фиктивного вызова вспомогательной функции вместо прагмы при разборе. Затем вы можете обнаружить эту функцию по имени в промежуточном представлении.

Кроме того, несколько дней назад в GCC ML был задан вопрос от felix.yang (huawei): "Как передать информацию о прагме, связанной с петлями, из TREE в RTL?" - http://comments.gmane.org/gmane.comp.gcc.devel/135243 - проверить ветку

Некоторые рекомендации из списка:

Посмотрите, как мы реализуем #pragma ivdep (см. Replace_loop_annotate () и fortran/trans-stmt.c, где он создает ANNOTATE_EXPR).

Патч с replace_loop_annotate() добавление функции и ivdep реализация прагмы: "Re: Patch: добавить поддержку #pragma ivdep для ME и C FE" Тобиаса Бернуса (2013-08-24).

Answer 2

Я не думаю, что регистрация прагмы DEFERRED в плагине возможна, поскольку обработчик отложенной прагмы не отображается на уровне плагина GCC.

Таким образом, ваша прагма работает только на этапе предварительной обработки, а не на этапе синтаксического анализа, поэтому достичь цели оптимизации довольно сложно.