TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

LayerNormalization failure of TensorRT 8.2.4.2

Open lin-lcx opened this issue 1 year ago • 3 comments

Description

Does 8.2.4.2 support LayerNormalization? With the same onnx model and conversion instructions, I successfully converted on 8.6.1.6, but not on 8.2.4.2. But I have to adapt to the 8.2.4.2 environment

[01/31/2024-10:15:34] [I] [TRT] No importer registered for op: LayerNormalization. Attempting to import as plugin.
[01/31/2024-10:15:34] [I] [TRT] Searching for plugin: LayerNormalization, plugin_version: 1, plugin_namespace: 
[01/31/2024-10:15:34] [E] [TRT] ModelImporter.cpp:773: While parsing node number 16 [LayerNormalization -> "/embeddings/norm/LayerNormalization_output_0"]:
[01/31/2024-10:15:34] [E] [TRT] ModelImporter.cpp:774: --- Begin node ---
[01/31/2024-10:15:34] [E] [TRT] ModelImporter.cpp:775: input: "/embeddings/patch_embeddings/Transpose_output_0"
input: "embeddings.norm.weight"
input: "embeddings.norm.bias"
output: "/embeddings/norm/LayerNormalization_output_0"
name: "/embeddings/norm/LayerNormalization"
op_type: "LayerNormalization"
attribute {
  name: "axis"
  i: -1
  type: INT
}
attribute {
  name: "epsilon"
  f: 1e-05
  type: FLOAT
}

Environment

TensorRT Version:8.2.4.2

NVIDIA GPU:2080ti

NVIDIA Driver Version:

CUDA Version:11.1

CUDNN Version:

lin-lcx avatar Jan 31 '24 02:01 lin-lcx

Could you please try latest 9.2/9.3 release? Thanks!

zerollzeng avatar Feb 01 '24 14:02 zerollzeng

Due to engineering environment restrictions, I must use the 8.2.4.2 environment. I am using trtexec

lin-lcx avatar Feb 02 '24 01:02 lin-lcx

IIRC most of the transformer support are added since TRT 8.6. And we won't back port feature back to old release.

zerollzeng avatar Feb 07 '24 09:02 zerollzeng

closing since no activity for more than 3 weeks, thanks all!

ttyio avatar Mar 05 '24 17:03 ttyio

I'm facing the same situation where I have to use TRT 8.2, may I know how you solved the problem?

ZeppelinSCB avatar Aug 25 '24 09:08 ZeppelinSCB