register float16

test=develop

register float16
test=develop
f7294f8b · Xin Pan · c132c790 · f7294f8b · f7294f8b
Showing with 5 addition and 2 deletion

paddle/fluid/operators/fill_constant_op.cc paddle/fluid/operators/fill_constant_op.cc +2 -1

paddle/fluid/operators/fill_constant_op.cu.cc paddle/fluid/operators/fill_constant_op.cu.cc +3 -1

未找到文件。
--- a/paddle/fluid/operators/fill_constant_op.cc
+++ b/paddle/fluid/operators/fill_constant_op.cc
@@ -87,4 +87,5 @@ REGISTER_OPERATOR(fill_constant, ops::FillConstantOp, ops::FillConstantOpMaker,
 REGISTER_OP_CPU_KERNEL(fill_constant, ops::FillConstantKernel<float>,
                       ops::FillConstantKernel<double>,
                       ops::FillConstantKernel<int64_t>,
-                       ops::FillConstantKernel<int>);
+                       ops::FillConstantKernel<int>,
+                       ops::FillConstantKernel<paddle::platform::float16>);
--- a/paddle/fluid/operators/fill_constant_op.cu.cc
+++ b/paddle/fluid/operators/fill_constant_op.cu.cc
@@ -17,4 +17,6 @@ limitations under the License. */
 namespace ops = paddle::operators;
 REGISTER_OP_CUDA_KERNEL(fill_constant, ops::FillConstantKernel<float>,
                        ops::FillConstantKernel<double>,
-                        ops::FillConstantKernel<int64_t>);
+                        ops::FillConstantKernel<int64_t>,
+                        ops::FillConstantKernel<int>,
+                        ops::FillConstantKernel<paddle::platform::float16>);