Skip to content

Commit 4dfcb29

Browse files
【CUDA Kernel No.7】fused_seqpool_cvm_grad算子Kernel修复 (#2008)
1 parent 8fa2140 commit 4dfcb29

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backends/iluvatar_gpu/kernels/cuda_kernels/fused_bias_residual_layernorm_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
// limitations under the License.
1414

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/fusion/gpu/fused_layernorm_kernel.cu" //NOLINT
16+
#include "paddle/phi/kernels/fused_layernorm_kernel.h"
1717

1818
PD_CUSTOM_KERNEL_REGISTER(fused_bias_residual_layernorm,
1919
iluvatar_gpu,

backends/metax_gpu/kernels/fusion/fused_layernorm_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,9 @@
1313
// limitations under the License.
1414

1515
#include "paddle/phi/core/kernel_registry.h"
16+
#include "paddle/phi/kernels/fused_layernorm_kernel.h"
1617
#include "paddle/phi/kernels/fusion/gpu/attention_layer.norm.h"
1718
#include "paddle/phi/kernels/fusion/gpu/fused_dropout_helper.h"
18-
#include "paddle/phi/kernels/fusion/gpu/fused_layernorm_kernel.cu" //NOLINT
1919

2020
#ifndef PADDLE_WITH_HIP
2121
#if CUDNN_VERSION_MIN(8, 1, 0)

0 commit comments

Comments
 (0)