for some complex schedule, how can I only lower one stage?
Thanks
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