Skip to content
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

Accept shape tensors in Compiler and propagate bounds #65

Open
jhalakpatel opened this issue Aug 8, 2024 · 1 comment
Open

Accept shape tensors in Compiler and propagate bounds #65

jhalakpatel opened this issue Aug 8, 2024 · 1 comment
Labels
tripy Pull request for the tripy project

Comments

@jhalakpatel
Copy link
Collaborator

We should accept shape tensors when setting up optimization profiles in the compiler. Additionally, the compiled function should provide some kind of mechanism to query shape bounds of its outputs.
@jhalakpatel jhalakpatel added enhancement New feature or request tripy Pull request for the tripy project labels Aug 8, 2024
@jhalakpatel jhalakpatel linked a pull request Aug 8, 2024 that will close this issue
Accept shape tensors in Compiler #64
Closed
@parthchadha
Copy link
Collaborator

This task also includes adding support for providing value bounds to the shape inputs.

@pranavm-nvidia pranavm-nvidia removed the enhancement New feature or request label Nov 26, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
tripy Pull request for the tripy project
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

Accept shape tensors in Compiler NVIDIA/TensorRT-Incubator
3 participants
@parthchadha @jhalakpatel @pranavm-nvidia