From bf0ef606ed5bf7d0ae7de8d65d945f9011046091 Mon Sep 17 00:00:00 2001 From: zhaoyingli <86812880+zhaoyinglia@users.noreply.github.com> Date: Tue, 15 Aug 2023 11:45:22 +0800 Subject: [PATCH] disable fast_layer_norm (#56263) --- paddle/phi/kernels/gpu/layer_norm_kernel.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/phi/kernels/gpu/layer_norm_kernel.cu b/paddle/phi/kernels/gpu/layer_norm_kernel.cu index c5bb0c288f2..9ba6dfd4235 100644 --- a/paddle/phi/kernels/gpu/layer_norm_kernel.cu +++ b/paddle/phi/kernels/gpu/layer_norm_kernel.cu @@ -606,7 +606,7 @@ void LayerNormKernel(const Context &dev_ctx, if ((feature_size >= 768 && feature_size <= 2048 && feature_size % 256 == 0 || feature_size == 4096) && scale != nullptr && bias != nullptr) { - can_call_fast_kernel = true; + can_call_fast_kernel = false; } if (can_call_fast_kernel) { -- GitLab