Skip to content

Commit 74569e3

Browse files
authored
Merge branch 'develop' into 82
2 parents 18b50eb + 64d769a commit 74569e3

File tree

5 files changed

+11
-12
lines changed

5 files changed

+11
-12
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: 3 additions & 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
@@ -300,6 +301,8 @@ file(
300301
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/broadcast_kernel.cu
301302
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/decayed_adagrad_kernel.cu
302303
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/debug_tools_kernel.cu
304+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/deformable_conv_kernel.cu
305+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/deformable_conv_grad_kernel.cu
303306
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/cumprod_grad_kernel.cu
304307
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/cumprod_kernel.cu
305308
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/crop_grad_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) {}

backends/metax_gpu/kernels/cuda_kernels/deformable_conv_grad_kernel_register.cu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,8 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
#include "paddle/phi/kernels/gpu/deformable_conv_grad_kernel.cu" // NOLINT
15+
#include "paddle/phi/core/kernel_registry.h"
16+
#include "paddle/phi/kernels/deformable_conv_grad_kernel.h"
1617

1718
PD_CUSTOM_KERNEL_REGISTER(deformable_conv_grad,
1819
metax_gpu,

backends/metax_gpu/kernels/cuda_kernels/deformable_conv_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/gpu/deformable_conv_kernel.cu" // NOLINT
16+
#include "paddle/phi/kernels/deformable_conv_kernel.h"
1717

1818
PD_CUSTOM_KERNEL_REGISTER(deformable_conv,
1919
metax_gpu,

0 commit comments

Comments
 (0)