TensorRT
TensorRT copied to clipboard
chore: Adapt CIA ops decomposition handling in upsample converters to torch 2.6
Description
Fixes #3226
Type of change
- Bug fix (non-breaking change which fixes an issue)
Checklist:
- [x] My code follows the style guidelines of this project (You can use the linters)
- [x] I have performed a self-review of my own code
- [ ] I have commented my code, particularly in hard-to-understand areas and hacks
- [ ] I have made corresponding changes to the documentation
- [ ] I have added tests to verify my fix or my feature
- [x] New and existing unit tests pass locally with my changes
- [ ] I have added the relevant labels to my PR in so that relevant reviewers are notified
Just saw a new update from torch decomp upstream, We need add that also from torch._export.utils import _decomp_table_to_post_autograd_aten