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

SkipLayerNorm plugin Error!!! #1919

Closed
DataXujing opened this issue Apr 12, 2022 · 3 comments
Closed

SkipLayerNorm plugin Error!!! #1919

DataXujing opened this issue Apr 12, 2022 · 3 comments
Labels
bug Something isn't working triaged Issue has been triaged by maintainers

Comments

@DataXujing
Copy link

When I use SkipLayerNorm Plugin in ONNX,and execute trtexec, I get the Error:

[04/12/2022-11:26:49] [I] [TRT] Searching for plugin: CustomSkipLayerNormPluginDynamic, plugin_version: 3, plugin_namespace:
[04/12/2022-11:26:49] [E] [TRT] SkipLayerNorm: invalid beta
[04/12/2022-11:26:49] [E] [TRT] SkipLayerNorm: invalid gamma
[04/12/2022-11:26:49] [F] [TRT] Assertion failed: mLd > 0
/home/jenkins/agent/workspace/OSS/OSS_L0_MergeRequest/oss/plugin/skipLayerNormPlugin/skipLayerNormInt8InterleavedPlugin.cpp:81
Aborting...

Aborted (core dumped)

```
@DataXujing
Copy link
Author

1649763683(1)

@ttyio ttyio added bug Something isn't working Topic: ONNX Plugin triaged Issue has been triaged by maintainers labels Apr 15, 2022
@ttyio
Copy link
Collaborator

ttyio commented Apr 15, 2022

Thanks @DataXujing , checked that the getFieldNames() implementation is not correct. Will fixed this, before we release the fix, you could WAR this by manually build the OSS after some patch like this (emplace all the necessary fields in mPluginAttributes):
https://github.com/NVIDIA/TensorRT/blob/main/plugin/fcPlugin/fcPlugin.cpp#L691

@DataXujing
Copy link
Author

OK,thanks for your reply!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working triaged Issue has been triaged by maintainers
Projects
None yet
Development

No branches or pull requests

2 participants