diff --git a/paddle/fluid/operators/slice_op.cc b/paddle/fluid/operators/slice_op.cc old mode 100755 new mode 100644 index 8d23ce8d11832c599ccfb7479a2ad388c91641c9..2b3be3d78b295104746f5fc07ffe0c97e09d37dd --- a/paddle/fluid/operators/slice_op.cc +++ b/paddle/fluid/operators/slice_op.cc @@ -460,6 +460,7 @@ REGISTER_OPERATOR(slice_grad, REGISTER_OP_CPU_KERNEL( slice, ops::SliceKernel, + ops::SliceKernel, ops::SliceKernel, ops::SliceKernel, ops::SliceKernel, @@ -471,6 +472,7 @@ REGISTER_OP_CPU_KERNEL( REGISTER_OP_CPU_KERNEL( slice_grad, ops::SliceGradKernel, + ops::SliceGradKernel, ops::SliceGradKernel, ops::SliceGradKernel, ops::SliceGradKernel, @@ -482,6 +484,7 @@ REGISTER_OP_CPU_KERNEL( REGISTER_OP_CUDA_KERNEL( slice, ops::SliceKernel, + ops::SliceKernel, ops::SliceKernel, ops::SliceKernel, ops::SliceKernel, @@ -494,6 +497,7 @@ REGISTER_OP_CUDA_KERNEL( REGISTER_OP_CUDA_KERNEL( slice_grad, ops::SliceGradKernel, + ops::SliceGradKernel, ops::SliceGradKernel, ops::SliceGradKernel, ops::SliceGradKernel, diff --git a/paddle/phi/kernels/cpu/slice_grad_kernel.cc b/paddle/phi/kernels/cpu/slice_grad_kernel.cc index 7e3efd217511f851efec2e616fe116b196cbda07..f22e3634f152485cf836bfa82f6c1eb653ceb57a 100644 --- a/paddle/phi/kernels/cpu/slice_grad_kernel.cc +++ b/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, diff --git a/paddle/phi/kernels/cpu/slice_kernel.cc b/paddle/phi/kernels/cpu/slice_kernel.cc index 0f2fe98a853235afb7ac27259a5ce7ff172edde8..ff9a5c1593fbad5affdfaaa75b30dd95717f7581 100644 --- a/paddle/phi/kernels/cpu/slice_kernel.cc +++ b/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, diff --git a/paddle/phi/kernels/funcs/eigen/pad.cc b/paddle/phi/kernels/funcs/eigen/pad.cc index c457199b0a93cae58f3f5ad32c8e62a445910c50..8041fc4ae175ec9e68e8a393e5d5cc5ef7180b39 100644 --- a/paddle/phi/kernels/funcs/eigen/pad.cc +++ b/paddle/phi/kernels/funcs/eigen/pad.cc @@ -59,6 +59,7 @@ struct EigenPad { template struct FUNCTOR; INSTANTIATION(EigenPad, bool); +INSTANTIATION(EigenPad, uint8_t); INSTANTIATION(EigenPad, int); INSTANTIATION(EigenPad, int64_t); INSTANTIATION(EigenPad, float); diff --git a/paddle/phi/kernels/funcs/eigen/pad.cu b/paddle/phi/kernels/funcs/eigen/pad.cu index 7d8c2580d962130d2d2f5de288e6bdb617155ba5..c4a3dd9ecc4f57907304391b0c4c34a51ba2c961 100644 --- a/paddle/phi/kernels/funcs/eigen/pad.cu +++ b/paddle/phi/kernels/funcs/eigen/pad.cu @@ -59,6 +59,7 @@ struct EigenPad { template struct FUNCTOR; \ template struct FUNCTOR INSTANTIATION(EigenPad, bool); +INSTANTIATION(EigenPad, uint8_t); INSTANTIATION(EigenPad, int); INSTANTIATION(EigenPad, int64_t); INSTANTIATION(EigenPad, float); diff --git a/paddle/phi/kernels/funcs/eigen/slice.cu b/paddle/phi/kernels/funcs/eigen/slice.cu index 36f705f168c81b742e52df90b76b8185b6b5b17a..ade58d069875980ec13aefd94b41381e937b4eba 100644 --- a/paddle/phi/kernels/funcs/eigen/slice.cu +++ b/paddle/phi/kernels/funcs/eigen/slice.cu @@ -59,6 +59,7 @@ struct EigenSlice { template struct FUNCTOR; \ template struct FUNCTOR INSTANTIATION(EigenSlice, bool); +INSTANTIATION(EigenSlice, uint8_t); INSTANTIATION(EigenSlice, int); INSTANTIATION(EigenSlice, int64_t); INSTANTIATION(EigenSlice, float); diff --git a/paddle/phi/kernels/gpu/slice_grad_kernel.cu.cc b/paddle/phi/kernels/gpu/slice_grad_kernel.cu.cc index a6db80abaee17ec2a7395ed7d4300ca1ff15a662..ec575ab952b05b4855ab1fc6e0963a69d8392002 100644 --- a/paddle/phi/kernels/gpu/slice_grad_kernel.cu.cc +++ b/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, diff --git a/paddle/phi/kernels/gpu/slice_kernel.cu.cc b/paddle/phi/kernels/gpu/slice_kernel.cu.cc index 8743163b220b9e0f917fdd805abfb935d68bc48f..5232ce35811c0d7f9257fd9f76c6bfcfcec093a2 100644 --- a/paddle/phi/kernels/gpu/slice_kernel.cu.cc +++ b/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,