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

请问动态 shape 下的空间如何申请呢？ #3

Open

hermosayhl opened this issue Aug 8, 2022 · 1 comment

hermosayhl commented Aug 8, 2022 •

edited

请问，大佬知道 dynamic shape 下的 getWorkspaceSize 如何写吗？静态的很好判断，但目前个人自定义的算子卡在了这里，还望大佬指点！

Member

DataXujing commented Sep 13, 2022

https://docs.nvidia.com/deeplearning/tensorrt/developer-guide/index.html#build_engine_c
我不确定您说的·getWorkspaceSize`是不是这个，我一般的做法是默认值，如果不行会在默认值的基础上增加，直到可以正常导出

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment