pytorch icon indicating copy to clipboard operation
pytorch copied to clipboard

Published 20 hours ago • verified publisher icon csarofeen

Transpose scheduler plan

Open zasdfgbnm opened this issue 3 years ago • 2 comments

Tracking the remaining things needed for the transpose scheduler, the order is the priority:

  • [x] Handle the case where the transposed dim is small (https://github.com/csarofeen/pytorch/pull/1910)
  • [x] More validation on broadcasting
  • [x] enable in registry
  • [ ] enable transpose in python (https://github.com/csarofeen/pytorch/pull/1882, thanks @rdspring1, https://github.com/pytorch/pytorch/pull/84629, thanks @IvanYashchuk )
  • [ ] More validation on trivial reduction
  • [ ] Per-tensor fine-grained schedule for group 2 (see comment https://github.com/csarofeen/pytorch/pull/1906#pullrequestreview-1071496777)
  • [ ] transpose + view

zasdfgbnm avatar Aug 12 '22 18:08 zasdfgbnm

Here is Ivan's PR to enable transpose with primTorch: https://github.com/pytorch/pytorch/pull/84629

rdspring1 avatar Sep 13 '22 17:09 rdspring1

Here is Ivan's PR to enable transpose with primTorch: pytorch#84629

Updated, thanks!

zasdfgbnm avatar Sep 13 '22 17:09 zasdfgbnm