From c5c3a99ebe496452c7c9d1a871a05ca5a1c9765d Mon Sep 17 00:00:00 2001 From: Zhan Rongrui <2742392377@qq.com> Date: Wed, 28 Feb 2024 02:27:56 +0000 Subject: [PATCH] fix --- python/paddle/incubate/nn/functional/fused_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/incubate/nn/functional/fused_transformer.py b/python/paddle/incubate/nn/functional/fused_transformer.py index 2dbeb2389f0f03..4d91f9b86f562b 100644 --- a/python/paddle/incubate/nn/functional/fused_transformer.py +++ b/python/paddle/incubate/nn/functional/fused_transformer.py @@ -413,7 +413,7 @@ def fused_bias_dropout_residual_layer_norm( x.shape[len(x.shape) - 1] == ln_bias.shape[0] ), "The dim of ln_bias must equal to the last dim of x." - if in_dynamic_mode(): + if in_dynamic_or_pir_mode(): if default_main_program().random_seed != 0: seed = default_main_program().random_seed (