feat: Decomposition for `_unsafe_index` by gs-olive · Pull Request #2386 · pytorch/TensorRT

@gs-olive

Description

  • Necessary decomposition for many models which use the unsafe index variant

Type of change

  • New feature (non-breaking change which adds functionality)

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
  • [ x ] I have commented my code, particularly in hard-to-understand areas and hacks
  • [ x ] 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
  • [ x ] I have added the relevant labels to my PR in so that relevant reviewers are notified

@gs-olive

github-actions[bot]

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code conforms to C++ style guidelines

github-actions[bot]

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code conforms to Python style guidelines

narendasan

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

gs-olive added a commit that referenced this pull request

Oct 10, 2023

@gs-olive