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

fix loading trt engine #1709

Merged
merged 1 commit into from
Oct 22, 2024
Merged

fix loading trt engine #1709

merged 1 commit into from
Oct 22, 2024

Conversation

hnyu
Copy link
Collaborator

@hnyu hnyu commented Oct 9, 2024

Currently, when loading a saved trt engine file, there is an error "getPluginCreator could not find plugin InstanceNormalization_TRT version 1" on some machines. Need to manually init libnvinfer before using tensorrt.

Copy link
Contributor

@Haichao-Zhang Haichao-Zhang left a comment

Choose a reason for hiding this comment

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

On which of the machines do we have this issue?

@hnyu
Copy link
Collaborator Author

hnyu commented Oct 22, 2024

On which of the machines do we have this issue?

hyaku laptop

@hnyu hnyu merged commit a56fe61 into pytorch Oct 22, 2024
2 checks passed
@hnyu hnyu deleted the PR_fix_trt_loading branch October 22, 2024 03:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants