Skip to content

Commit 982ed6f

Browse files
Le-soleileYqGe585
andauthored
【CUDA Kernel No.44】算子Kernel修复 (#2237)
* fix * The original code naming convention has not been updated --------- Co-authored-by: Yuqiang Ge <[email protected]>
1 parent d5172d9 commit 982ed6f

File tree

3 files changed

+6
-10
lines changed

3 files changed

+6
-10
lines changed

backends/iluvatar_gpu/kernels/cuda_kernels/ctc_align_kernel_register.cu

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,7 @@
1313
// limitations under the License.
1414

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

18-
PD_CUSTOM_KERNEL_REGISTER(ctc_align,
19-
iluvatar_gpu,
20-
ALL_LAYOUT,
21-
phi::CTCAlignOpCUDAKernel,
22-
int,
23-
int64_t) {}
18+
PD_CUSTOM_KERNEL_REGISTER(
19+
ctc_align, iluvatar_gpu, ALL_LAYOUT, phi::CTCAlignKernel, int, int64_t) {}

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,7 @@ file(
124124
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/math/*.cu
125125
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/eigen/*.cu
126126
# kernels/gpu
127+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/ctc_align_kernel.cu
127128
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/yolo_box_post_kernel.cu
128129
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_grad_kernel.cu
129130
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/ctc_align_kernel_register.cu

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,7 @@
1313
// limitations under the License.
1414

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

1818
PD_CUSTOM_KERNEL_REGISTER(
19-
ctc_align, metax_gpu, ALL_LAYOUT, phi::CTCAlignOpCUDAKernel, int, int64_t) {
20-
}
19+
ctc_align, metax_gpu, ALL_LAYOUT, phi::CTCAlignKernel, int, int64_t) {}

0 commit comments

Comments
 (0)