我正在创建一个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)。我正在做我的毕业论文(自己的选择),欢迎任何志同道合的人。