GCC插件,添加新的优化指令

13

我正在创建一个GCC插件。

我试图创建一个针对特定循环转换的插件,即将循环完全展开N次(给定参数)。我已经正确安装了插件,并且可以在编译过程中成功注册我的pragma。当我使用函数c_register_pragma注册pragma时,我可以在词法分析中处理它(使用函数handle_my_pragma),但是之后我该如何找到它呢?

我也可以定义自己的pass并遍历GIMPLE,但是没有任何pragma的痕迹。所以我的问题是:我的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)。我正在做我的毕业论文(自己的选择),欢迎任何志同道合的人。


来自Broadcom的梅冰峰为GCC提供了“pragma unroll”补丁- http://gcc.gnu.org/ml/gcc/2008-05/msg00426.html PS 请问您使用的是哪个版本的GCC? - osgx
2个回答

3
当我使用函数c_register_pragma注册pragma时,我可以在词法分析中处理它(使用函数handle_my_pragma),但接下来我该怎么找到它呢?
有一个选项(实际上是hack),在解析时在pragma的位置创建虚构的辅助函数调用。然后你可以在中间表示中通过名称检测到这个函数。
另外,几天前,在GCC ML上有一个问题,来自华为的felix.yang:“如何从树形结构传递与循环相关的pragma信息到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编译指示实现的补丁:"回复:补丁:将#pragma ivdep支持添加到ME和C FE中",作者为Tobias Burnus(2013-08-24)。

嘿,感谢你的回答!关于这个替换“hack”,我在哪里可以找到相关信息?如何使用它?我找不到有关pragma ivdep实现的任何信息,因此这种替换看起来是可用的。不幸的是,我没有太多时间了,所以我需要使用更快的方法 :/ - Michal Bukovy
Michal,有一个ivdep补丁(使用“replace_loop_annotate”进行谷歌搜索):http://gcc.gnu.org/ml/gcc-patches/2013-10/msg02031.html -- 如果有ivdep循环,在c_parser_for_statement中调用代码build2(ANNOTATE_EXPR..。还有一个文本添加到extend.texi - http://gcc.gnu.org/onlinedocs/gcc/Loop-Specific-Pragmas.html - osgx
Michal,你只需要“GCC插件”,还是允许你对GCC代码库进行修补? - osgx
我更倾向于避免打补丁。请问,你能告诉我如何用函数替换#pragma吗? 不过,非常感谢你的帮助! - Michal Bukovy

0

我认为在插件中注册DEFERRED pragma是不可能的,因为延迟pragma的处理程序在GCC插件级别中没有暴露出来。

因此,你的pragma只能在预处理阶段起作用,而不是解析阶段,这样就很难实现优化目标了。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接