c++ - GCC 插件，添加新的优化编译指示

Question

我正在创建一个 GCC 插件。

我正在尝试为特定的循环转换创建一个插件 - 展开循环恰好 N(给定参数)次。我已经正确安装了插件，并且可以在编译过程中成功注册我的编译指示。当我使用函数 c_register_pragma 注册 pragma 时，我可以在词法分析中处理它(使用函数 handle_my_pragma)，但是我怎样才能找到它呢？

我也可以定义自己的pass和traverse GIMPLE，但是没有任何pragma的痕迹。所以我的问题是:我的编译指示在哪里，我如何影响我的代码？或者你会建议什么来达到我的目标？它不必与 pragma 一起使用，但这似乎是一个好主意。另外，我知道 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

When I register pragma with function c_register_pragma, I can handle it in lexical analysis (with function handle_my_pragma), but how can I find it then?

在解析时，有一个选项(实际上是 hack)可以在 pragma 的位置创建虚构的辅助函数调用。然后您可以在中间表示中按名称检测此函数。

Aslo，前几天在 felix.yang(华为)的 GCC ML 中有一个问题“如何将循环相关的 pragma 信息从 TREE 传递到 RTL？” - http://comments.gmane.org/gmane.comp.gcc.devel/135243 - 检查线程

列表中的一些建议:

Look at how we implement #pragma ivdep (see replace_loop_annotate () and fortran/trans-stmt.c where it builds ANNOTATE_EXPR).

带有 replace_loop_annotate() 函数添加和 ivdep pragma 实现的补丁:"Re: Patch: Add #pragma ivdep support to the ME and C FE"作者:Tobias Burnus (2013-08-24)。