I just realized that this problem is even more complicate. If we simply rename 
the second `T_reshape` to `T_reshape_1`, then you still cannot match the TE 
compute unless we rename the corresponding one in the compute DAG as well. 
However, printing the compute DAG and schedule are different functions and it 
would be unsafe to maintain a such mapping. While the meta schedule that 
@junrushao1994 et all are working on has resolved this issue, I would say the 
best way for now is probably just to manually match them, as we know that the 
line with `... = tuple(name.op.axis) + tuple(name.op.reduce_axis)` is always 
the starting point of scheduling a stage.





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/print-auto-schedule-python-schedule-with-topi-op/11363/5)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/3d981efb9aacdf78612861e679afba89b8383a7feb2294d27a5eb1f68909de34).

Reply via email to