Incorrect Boundary Infer when scheduling contain tvm_if_then_else

[aca88]

Hello,

I also encountered problems as the ones you described in your post by building schedules with more than one conv2d with padding = 'same'. I opened a new topic [TVM Scheduling] Split factors missmatch to original extent - Questions - Apache TVM Discuss and therefore would like to know more about your fix

@wezuo

Currently, we made a fix locally to use the intersect of the condition in the if_then_else with the iteration variable domain, to infer the output tensor. But I would really like to hear about your solutions.

1. Is there an ETA available for this PR?
2. Where exactly did you do the change? is it inside the InferBounds routine or is it an added ir_pass which makes changes based on the AST?
3. Also I was wondering if the change could work also for the case from my post copied below for ease of lecture:

@aca88

Suppose now that you would want to split some axis and you wish that the split factors are integer dividers of the original domains (i.e. if(not some_tvm_axis.dom.extent%16){some_factor=4} ) . Because the methods mentioned before (i.e. InferBounds) change the domains of the axes of all stages, then you can no longer expect the inferred bounds to still be an integer multiplier of the factors.

Thanks