Skip to content

↔ [Converter] Add support for aten::Int.Tensor in Torch-TensorRT TorchScript Path #1836

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Closed
gs-olive opened this issue Apr 18, 2023 · 3 comments · Fixed by #1937
Closed

↔ [Converter] Add support for aten::Int.Tensor in Torch-TensorRT TorchScript Path #1836

gs-olive opened this issue Apr 18, 2023 · 3 comments · Fixed by #1937
Assignees
@gs-olive
Labels
component: converters Issues re: Specific op converters feature request New feature or request

Comments

@gs-olive
Copy link
Collaborator

aten::Int.Tensor

  • Function Schema: aten::Int.Tensor(Tensor a) -> int

  • Original PyTorch API: torch.item() [documentation]

This converter/evaluator extracts the integer value from single-element tensors.
@gs-olive gs-olive added feature request New feature or request component: converters Issues re: Specific op converters labels Apr 18, 2023
@narendasan
Copy link
Collaborator

Is this for FX or TS?

@gs-olive
Copy link
Collaborator Author

This is for the TorchScript path and relates to a model for which this is the only missing operator.

@gs-olive gs-olive changed the title ↔ [Converter] Add support for aten::Int.Tensor in Torch-TensorRT ↔ [Converter] Add support for aten::Int.Tensor in Torch-TensorRT TorchScript Apr 19, 2023
@gs-olive gs-olive changed the title ↔ [Converter] Add support for aten::Int.Tensor in Torch-TensorRT TorchScript ↔ [Converter] Add support for aten::Int.Tensor in Torch-TensorRT TorchScript Path Apr 19, 2023
@xunleiw
Copy link

xunleiw commented Apr 19, 2023

The related issue closed more than 1 yr ago.
#513

@gs-olive gs-olive assigned gs-olive and unassigned peri044 Apr 19, 2023
@gs-olive gs-olive mentioned this issue Apr 19, 2023
Closed
@gs-olive gs-olive mentioned this issue May 2, 2023
Closed
@gs-olive gs-olive mentioned this issue May 19, 2023
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@gs-olive gs-olive

Labels
component: converters Issues re: Specific op converters feature request New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

fix/feat: Add lowering pass to resolve most aten::Int.Tensor uses gs-olive/TensorRT
4 participants
@narendasan @peri044 @xunleiw @gs-olive