Hello, is there a way to add pragmas in TIR schedules? As far as I saw annotate* primitive does not appear in lowered schedules.
*: sch.annotate(global_A_block, “ann_key”, “ann_value”)
Hello, is there a way to add pragmas in TIR schedules? As far as I saw annotate* primitive does not appear in lowered schedules.
*: sch.annotate(global_A_block, “ann_key”, “ann_value”)
In the lowering phase currently, all keys with “pragma_” prefix are converted to attribute nodes, just the same with TE-schedule, otherwise:
It worked, thanks.
so basic implementation would be as in following
global_A_block = sch.get_block(“A_global”)
lp =sch.get_loops(global_A_block)
sch.annotate(lp[0], “pragma_key”, “pragma_value”) # annotated on only first iteration variable of the nest loops