Skip to content

Commit 18043ab

Browse files
authored
fix the but that concat op can't support uint8 (#33667)
1 parent 9a3d859 commit 18043ab

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

paddle/fluid/operators/concat_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,4 +244,4 @@ REGISTER_OP_CPU_KERNEL(
244244
ops::ConcatGradKernel<paddle::platform::CPUDeviceContext,
245245
paddle::platform::float16>,
246246
ops::ConcatGradKernel<paddle::platform::CPUDeviceContext, int>,
247-
ops::ConcatKernel<paddle::platform::CPUDeviceContext, uint8_t>);
247+
ops::ConcatGradKernel<paddle::platform::CPUDeviceContext, uint8_t>);

paddle/fluid/operators/concat_op.cu.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,4 +33,4 @@ REGISTER_OP_CUDA_KERNEL(
3333
ops::ConcatGradKernel<paddle::platform::CUDADeviceContext, plat::float16>,
3434
ops::ConcatGradKernel<paddle::platform::CUDADeviceContext, int64_t>,
3535
ops::ConcatGradKernel<paddle::platform::CUDADeviceContext, int>,
36-
ops::ConcatKernel<paddle::platform::CUDADeviceContext, uint8_t>);
36+
ops::ConcatGradKernel<paddle::platform::CUDADeviceContext, uint8_t>);

0 commit comments

Comments
 (0)