Add support in cast op kernel

4785c00e · fengjiayi · 9f81f1ce · 4785c00e · 4785c00e
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/operators/cast_op.cc paddle/fluid/operators/cast_op.cc +1 -0

paddle/fluid/operators/cast_op.cu paddle/fluid/operators/cast_op.cu +1 -1

未找到文件。
--- a/paddle/fluid/operators/cast_op.cc
+++ b/paddle/fluid/operators/cast_op.cc
@@ -89,4 +89,5 @@ REGISTER_OP_CPU_KERNEL(cast, ops::CastOpKernel<CPU, float>,
                       ops::CastOpKernel<CPU, int>,
                       ops::CastOpKernel<CPU, int64_t>,
                       ops::CastOpKernel<CPU, bool>,
+                       ops::CastOpKernel<CPU, uint8_t>,
                       ops::CastOpKernel<CPU, paddle::platform::float16>);
--- a/paddle/fluid/operators/cast_op.cu
+++ b/paddle/fluid/operators/cast_op.cu
@@ -21,5 +21,5 @@ using CastOpKernel =

 REGISTER_OP_CUDA_KERNEL(cast, CastOpKernel<float>, CastOpKernel<double>,
                        CastOpKernel<int>, CastOpKernel<int64_t>,
-                        CastOpKernel<bool>,
+                        CastOpKernel<bool>, CastOpKernel<uint8_t>,
                        CastOpKernel<paddle::platform::float16>);