fix the but that concat op can't support uint8 (#33667)

18043ab5 · 李季 · GitHub · 9a3d8593 · 18043ab5 · 18043ab5
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

paddle/fluid/operators/concat_op.cc paddle/fluid/operators/concat_op.cc +1 -1

paddle/fluid/operators/concat_op.cu.cc paddle/fluid/operators/concat_op.cu.cc +1 -1

未找到文件。
--- a/paddle/fluid/operators/concat_op.cc
+++ b/paddle/fluid/operators/concat_op.cc
@@ -244,4 +244,4 @@ REGISTER_OP_CPU_KERNEL(
    ops::ConcatGradKernel<paddle::platform::CPUDeviceContext,
                          paddle::platform::float16>,
    ops::ConcatGradKernel<paddle::platform::CPUDeviceContext, int>,
-    ops::ConcatKernel<paddle::platform::CPUDeviceContext, uint8_t>);
+    ops::ConcatGradKernel<paddle::platform::CPUDeviceContext, uint8_t>);
--- a/paddle/fluid/operators/concat_op.cu.cc
+++ b/paddle/fluid/operators/concat_op.cu.cc
@@ -33,4 +33,4 @@ REGISTER_OP_CUDA_KERNEL(
    ops::ConcatGradKernel<paddle::platform::CUDADeviceContext, plat::float16>,
    ops::ConcatGradKernel<paddle::platform::CUDADeviceContext, int64_t>,
    ops::ConcatGradKernel<paddle::platform::CUDADeviceContext, int>,
-    ops::ConcatKernel<paddle::platform::CUDADeviceContext, uint8_t>);
+    ops::ConcatGradKernel<paddle::platform::CUDADeviceContext, uint8_t>);