Skip to content

Commit

Permalink
disable fast_layer_norm (#56263)
Browse files Browse the repository at this point in the history
  • Loading branch information
zhaoyinglia authored Aug 15, 2023
1 parent c27bd04 commit bf0ef60
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion paddle/phi/kernels/gpu/layer_norm_kernel.cu
Original file line number Diff line number Diff line change
Expand Up @@ -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) {
Expand Down

0 comments on commit bf0ef60

Please sign in to comment.