Skip to content

Commit 5c7ad3b

Browse files
author
Pei Yang
authored
remove skip-ln-plugin-restriction on 2.0 (#32119)
1 parent 97cd708 commit 5c7ad3b

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

paddle/fluid/inference/tensorrt/plugin/skip_layernorm_op_plugin.cu

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -54,11 +54,6 @@ void SkipLayerNormPluginDynamic::terminate() {
5454
nvinfer1::DimsExprs SkipLayerNormPluginDynamic::getOutputDimensions(
5555
int output_index, const nvinfer1::DimsExprs *inputs, int nb_inputs,
5656
nvinfer1::IExprBuilder &expr_builder) {
57-
PADDLE_ENFORCE_EQ(
58-
inputs[0].nbDims, 5,
59-
platform::errors::InvalidArgument(
60-
"The Input dim of the SkipLayernorm should be 5, but it's (%d) now.",
61-
inputs[0].nbDims));
6257
return inputs[0];
6358
}
6459

0 commit comments

Comments
 (0)