diff --git a/paddle/fluid/operators/cast_op.h b/paddle/fluid/operators/cast_op.h index 68d0832ba574a47c6c81de68970990d4e21a9e64..6220e57f5941d89cbf0aea268b85ad55af6132cc 100644 --- a/paddle/fluid/operators/cast_op.h +++ b/paddle/fluid/operators/cast_op.h @@ -55,7 +55,8 @@ class CastOpKernel : public framework::OpKernel { auto* in = context.Input("X"); auto* out = context.Output("Out"); framework::VisitDataType( - static_cast(context.Attr("out_dtype")), + static_cast( + context.Attr("out_dtype")), CastOpFunctor( in, out, context.template device_context())); }