From 270efb9693f7b691f842c696939ecedafc9f0622 Mon Sep 17 00:00:00 2001 From: duanboqiang Date: Thu, 26 Aug 2021 19:49:20 +0800 Subject: [PATCH] fix assign bug support fp16 uint8 (#35153) * fix assign bug support fp16 uint8 * fix dygragh assign bool bug * modify code style * revoke bool modification --- paddle/fluid/operators/assign_op.cc | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/paddle/fluid/operators/assign_op.cc b/paddle/fluid/operators/assign_op.cc index 433cabcfee0..09452b8f68b 100644 --- a/paddle/fluid/operators/assign_op.cc +++ b/paddle/fluid/operators/assign_op.cc @@ -160,15 +160,15 @@ REGISTER_OPERATOR(assign, ops::AssignOp, REGISTER_OP_CPU_KERNEL_FUNCTOR(assign, float, ops::AssignKernel, double, ops::AssignKernel, int, ops::AssignKernel, - int64_t, ops::AssignKernel, bool, - ops::AssignKernel, plat::float16, - ops::AssignKernel, plat::bfloat16, + int64_t, ops::AssignKernel, uint8_t, + ops::AssignKernel, bool, ops::AssignKernel, + plat::float16, ops::AssignKernel, plat::bfloat16, ops::AssignKernel); #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) REGISTER_OP_CUDA_KERNEL_FUNCTOR(assign, float, ops::AssignKernel, double, ops::AssignKernel, int, ops::AssignKernel, - int64_t, ops::AssignKernel, bool, - ops::AssignKernel, plat::float16, - ops::AssignKernel); + int64_t, ops::AssignKernel, uint8_t, + ops::AssignKernel, bool, ops::AssignKernel, + plat::float16, ops::AssignKernel); #endif -- GitLab