TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

chore: Adapt CIA ops decomposition handling in upsample converters to torch 2.6

Open HolyWu opened this issue 1 year ago • 1 comments

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

HolyWu avatar Oct 11 '24 16:10 HolyWu

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

lanluo-nvidia avatar Oct 20 '24 18:10 lanluo-nvidia