You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
fix: Add check to ensure einsum converter has no more than 2 tensor inputs (#1439)
* fix: Add check to ensure einsum converter has correct args
- TRT einsum implementation currently supports 2 inputs, however the
converter will accept any number of inputs and TRT throws an error at
compilation
- `aten::einsum` converter now checks that the tensor argument list does
not exceed 2 elements, and throws an informative error otherwise
* Add escaped quotes so user can copy-paste printed solution
0 commit comments