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.

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

Disable/Enable graph level optimizations #4275

Closed
banu-narayana-dnjp opened this issue Dec 10, 2024 · 4 comments
Closed

Disable/Enable graph level optimizations #4275

banu-narayana-dnjp opened this issue Dec 10, 2024 · 4 comments
Labels
Enhancement New feature or request triaged Issue has been triaged by maintainers

Comments

@banu-narayana-dnjp
Copy link

I have noticed that the several graph optimizations like operator fusions, etc are occured by default while inferencing in TensorRT.
Is there any way to disable them or does TRT have levels of graph ops just as in ONNXRuntime?
Because I want to compare the performaces between before opt and after opt, for ex.

@lix19937
Copy link

lix19937 commented Dec 11, 2024

mark all with polygraphy tool.

@asfiyab-nvidia asfiyab-nvidia added Enhancement New feature or request triaged Issue has been triaged by maintainers labels Dec 16, 2024
@asfiyab-nvidia
Copy link
Collaborator

asfiyab-nvidia commented Dec 16, 2024

@banu-narayana-dnjp Unfortunately, TensorRT API doesn't support disabling fusions at the moment.
However, you can experiment with the builder optimization level to achieve different fusion patterns. Please refer to doc on optimization level for usage

@banu-narayana-dnjp
Copy link
Author

Thank you @asfiyab-nvidia and @lix19937 for your clarifications. Just one question, does MARK_ALL with the polygraphy tool result in any change in the final score(considering it prevents the fusions)?

@lix19937
Copy link

Mainly to prevents the fusions.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Enhancement New feature or request triaged Issue has been triaged by maintainers
Projects
None yet
Development

No branches or pull requests

3 participants