我正在创建一个 GCC 插件。
我正在尝试为特定的循环转换创建一个插件 - 精确地展开循环 N(给定参数)次。我已经正确安装了插件,并且可以在编译过程中成功注册我的编译指示。当我用 function 注册 pragma 时c_register_pragma
,我可以在词法分析中处理它(用 function handle_my_pragma
),但是我怎样才能找到它呢?
我也可以定义自己的 pass 并遍历 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)。我这样做是因为我的最终论文(自己的选择),我欢迎任何灵魂伴侣。