How can tvm lower only one stage?

1

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

Thanks

2

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