-
Notifications
You must be signed in to change notification settings - Fork 7.2k
Add torch.compile support for all ops #8055
Copy link
Copy link
Closed
Description
We should add support for torch.compile in all of our ops (meta implem, symint support, etc.) and use the opcheck test suite. See https://docs.google.com/document/d/1W--T6wz8IY8fOI0Vm8BF44PdBgs283QvpelJZWieQWQ/edit#heading=h.pnrqcv6bkfn3 for more details.
- nms: Meta implementation for nms #7944 Add opcheck testing for nms #7961
- RoiAlign https://github.com/pytorch/vision/pull Add opchecks for RoiAlign #8057
- PSRoiAlign PSRoiAlign: SymInt support + meta-implem #8058
- RoiPool RoIPool: SymInt support + meta-implem #8061
- PSRoiPool PSRoiPool: SymInt support + meta-implem #8062
- DeformConv2D DeformConv2d: SymInt support + meta-implem + opchecks #8063
cc @pmeier
Reactions are currently unavailable