disable fast_layer_norm (#56263)

bf0ef606 · zhaoyingli · GitHub · c27bd049 · bf0ef606
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/phi/kernels/gpu/layer_norm_kernel.cu paddle/phi/kernels/gpu/layer_norm_kernel.cu +1 -1

未找到文件。
--- 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) {