How can tvm lower only one stage?

for some complex schedule, how can I only lower one stage?

Thanks

You may find

s[op].pragma(axis, 'debug_skip_region') 

helpful.

Then TVM will skip all the code below the scope of axis.
You can use it to skip any stages or for loops

1 Like