forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 7
Open
Description
Tracking the remaining things needed for the transpose scheduler, the order is the priority:
- Handle the case where the transposed dim is small (Transpose scheduler small dim sizes better support #1910)
- More validation on broadcasting
- enable in registry
- enable transpose in python (Enable Transpose operation #1882, thanks @rdspring1, Add nvFuser support for transpose pytorch/pytorch#84629, thanks @IvanYashchuk )
- More validation on trivial reduction
- Per-tensor fine-grained schedule for group 2 (see comment Improve view support on pointwise and transpose scheduler #1906 (review))
- transpose + view
Metadata
Metadata
Assignees
Labels
No labels