I read this post Adding a pragma onto a schedule and have a slightly different use case.
I want to put a pragma on a stage. Every leaf IterVar in the stage is tensorized. But I still want to use pragma to annotate the stage with some information. What’s the best way to do that?
I may also need a pragma for the entire schedule to instruct the compiler to run experimental implementation.
My current workaround is add these information to another stage’s leaf IterVar. Any better solution is appreicated.