slice op supports uint8_t (#47067)

paddle/fluid/operators/slice_op.cc

@@ -460,6 +460,7 @@ REGISTER_OPERATOR(slice_grad,
 REGISTER_OP_CPU_KERNEL(
     slice,
     ops::SliceKernel<phi::CPUContext, bool>,
+    ops::SliceKernel<phi::CPUContext, uint8_t>,
     ops::SliceKernel<phi::CPUContext, int>,
     ops::SliceKernel<phi::CPUContext, int64_t>,
     ops::SliceKernel<phi::CPUContext, float>,
@@ -471,6 +472,7 @@ REGISTER_OP_CPU_KERNEL(
 REGISTER_OP_CPU_KERNEL(
     slice_grad,
     ops::SliceGradKernel<phi::CPUContext, bool>,
+    ops::SliceGradKernel<phi::CPUContext, uint8_t>,
     ops::SliceGradKernel<phi::CPUContext, int>,
     ops::SliceGradKernel<phi::CPUContext, int64_t>,
     ops::SliceGradKernel<phi::CPUContext, float>,
@@ -482,6 +484,7 @@ REGISTER_OP_CPU_KERNEL(
 REGISTER_OP_CUDA_KERNEL(
     slice,
     ops::SliceKernel<phi::GPUContext, bool>,
+    ops::SliceKernel<phi::GPUContext, uint8_t>,
     ops::SliceKernel<phi::GPUContext, float>,
     ops::SliceKernel<phi::GPUContext, double>,
     ops::SliceKernel<phi::GPUContext, int>,
@@ -494,6 +497,7 @@ REGISTER_OP_CUDA_KERNEL(
 REGISTER_OP_CUDA_KERNEL(
     slice_grad,
     ops::SliceGradKernel<phi::GPUContext, bool>,
+    ops::SliceGradKernel<phi::GPUContext, uint8_t>,
     ops::SliceGradKernel<phi::GPUContext, float>,
     ops::SliceGradKernel<phi::GPUContext, double>,
     ops::SliceGradKernel<phi::GPUContext, int>,

paddle/phi/kernels/cpu/slice_grad_kernel.cc

@@ -23,6 +23,7 @@ PD_REGISTER_KERNEL(slice_grad,
                    ALL_LAYOUT,
                    phi::SliceGradRawKernel,
                    bool,
+                   uint8_t,
                    int,
                    int64_t,
                    float,

paddle/phi/kernels/cpu/slice_kernel.cc

@@ -23,6 +23,7 @@ PD_REGISTER_KERNEL(slice,
                    ALL_LAYOUT,
                    phi::SliceRawKernel,
                    bool,
+                   uint8_t,
                    int,
                    int64_t,
                    float,

paddle/phi/kernels/funcs/eigen/pad.cc

@@ -59,6 +59,7 @@ struct EigenPad<Eigen::DefaultDevice, T, Rank> {
   template struct FUNCTOR<Eigen::DefaultDevice, TYPE, 6>;
 
 INSTANTIATION(EigenPad, bool);
+INSTANTIATION(EigenPad, uint8_t);
 INSTANTIATION(EigenPad, int);
 INSTANTIATION(EigenPad, int64_t);
 INSTANTIATION(EigenPad, float);

paddle/phi/kernels/funcs/eigen/pad.cu

@@ -59,6 +59,7 @@ struct EigenPad<Eigen::GpuDevice, T, Rank> {
   template struct FUNCTOR<Eigen::GpuDevice, TYPE, 5>; \
   template struct FUNCTOR<Eigen::GpuDevice, TYPE, 6>
 INSTANTIATION(EigenPad, bool);
+INSTANTIATION(EigenPad, uint8_t);
 INSTANTIATION(EigenPad, int);
 INSTANTIATION(EigenPad, int64_t);
 INSTANTIATION(EigenPad, float);

paddle/phi/kernels/funcs/eigen/slice.cu

@@ -59,6 +59,7 @@ struct EigenSlice<Eigen::GpuDevice, T, Rank> {
   template struct FUNCTOR<Eigen::GpuDevice, TYPE, 5>; \
   template struct FUNCTOR<Eigen::GpuDevice, TYPE, 6>
 INSTANTIATION(EigenSlice, bool);
+INSTANTIATION(EigenSlice, uint8_t);
 INSTANTIATION(EigenSlice, int);
 INSTANTIATION(EigenSlice, int64_t);
 INSTANTIATION(EigenSlice, float);

paddle/phi/kernels/gpu/slice_grad_kernel.cu.cc

@@ -23,6 +23,7 @@ PD_REGISTER_KERNEL(slice_grad,
                    ALL_LAYOUT,
                    phi::SliceGradRawKernel,
                    bool,
+                   uint8_t,
                    int,
                    int64_t,
                    float,

paddle/phi/kernels/gpu/slice_kernel.cu.cc

@@ -23,6 +23,7 @@ PD_REGISTER_KERNEL(slice,
                    ALL_LAYOUT,
                    phi::SliceRawKernel,
                    bool,
+                   uint8_t,
                    int,
                    int64_t,
                    float,