Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[TE] Correctly generate buffer binds with axis separators (apache#10819)
In SchedulePostProcToPrimfunc, when the axis separator attribute is moved to the buffer properties, it doesn't update buffers that are in the buffer bind scope. This occurs if `Stage.tensorize` is called for a stage whose layout transformation includes `te.AXIS_SEPARATOR`.
- Loading branch information