Ah you’re right. In this case auto-scheduler doesn’t differentiate reshape ops, because their names are defined in the TOPI compute (and are the same…). I had this issue before but it was for cache_read
, which was added by the auto-scheduler so I could let auto-scheduler control its name.
We may need to improve the printer here (https://github.com/apache/tvm/blob/main/src/auto_scheduler/compute_dag.cc#L1206) by adding a renaming logic. I’ll try to take a look if I got time in this week.