fill_constant_batch_size_like op support fp16 (#39907)

b8cf8ca7 · WangXi · GitHub · f77019a0 · b8cf8ca7
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/operators/fill_constant_batch_size_like_op.cu.cc ...le/fluid/operators/fill_constant_batch_size_like_op.cu.cc +2 -0

未找到文件。
--- a/paddle/fluid/operators/fill_constant_batch_size_like_op.cu.cc
+++ b/paddle/fluid/operators/fill_constant_batch_size_like_op.cu.cc
@@ -18,6 +18,8 @@ limitations under the License. */
 namespace ops = paddle::operators;
 REGISTER_OP_CUDA_KERNEL(
    fill_constant_batch_size_like,
+    ops::FillConstantBatchSizeLikeOpKernel<paddle::platform::CUDADeviceContext,
+                                           paddle::platform::float16>,
    ops::FillConstantBatchSizeLikeOpKernel<paddle::platform::CUDADeviceContext,
                                           float>,
    ops::FillConstantBatchSizeLikeOpKernel<paddle::platform::CUDADeviceContext,