From edd66f2e225f54fc6193b94d8ab71343015e6c91 Mon Sep 17 00:00:00 2001 From: Sing_chan <51314274+betterpig@users.noreply.github.com> Date: Thu, 25 Aug 2022 19:13:55 +0800 Subject: [PATCH] make full_like support double_max in dygraph (#45385) * make full_like support double_max in dygraph * fix bug --- paddle/fluid/pybind/eager_utils.cc | 2 +- paddle/phi/kernels/cpu/full_kernel.cc | 2 +- paddle/phi/kernels/gpu/full_kernel.cu | 2 +- paddle/phi/kernels/xpu/full_kernel.cc | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/paddle/fluid/pybind/eager_utils.cc b/paddle/fluid/pybind/eager_utils.cc index 147154ae549..62a28868bb8 100644 --- a/paddle/fluid/pybind/eager_utils.cc +++ b/paddle/fluid/pybind/eager_utils.cc @@ -1289,7 +1289,7 @@ paddle::experimental::Scalar CastPyArg2Scalar(PyObject* obj, int64_t value = CastPyArg2Long(obj, op_type, arg_pos); return paddle::experimental::Scalar(value); } else if (PyFloat_Check(obj)) { - float value = CastPyArg2Float(obj, op_type, arg_pos); + double value = CastPyArg2Double(obj, op_type, arg_pos); return paddle::experimental::Scalar(value); } else if (IsEagerTensor(obj)) { paddle::experimental::Tensor& value = GetTensorFromPyObject( diff --git a/paddle/phi/kernels/cpu/full_kernel.cc b/paddle/phi/kernels/cpu/full_kernel.cc index ceb2312b53a..6571cb2ca8f 100644 --- a/paddle/phi/kernels/cpu/full_kernel.cc +++ b/paddle/phi/kernels/cpu/full_kernel.cc @@ -44,7 +44,7 @@ void FullLikeKernel(const Context& dev_ctx, const Scalar& val, DataType dtype, DenseTensor* out) { - auto value = val.to(); + auto value = val.to(); using CommonType = typename std::common_type< float, typename std::conditional::value, diff --git a/paddle/phi/kernels/gpu/full_kernel.cu b/paddle/phi/kernels/gpu/full_kernel.cu index 50e57a46317..684e9156609 100644 --- a/paddle/phi/kernels/gpu/full_kernel.cu +++ b/paddle/phi/kernels/gpu/full_kernel.cu @@ -60,7 +60,7 @@ void FullLikeKernel(const Context& dev_ctx, const Scalar& val, DataType dtype, DenseTensor* out) { - auto value = val.to(); + auto value = val.to(); using CommonType = typename std::common_type< float, typename std::conditional< diff --git a/paddle/phi/kernels/xpu/full_kernel.cc b/paddle/phi/kernels/xpu/full_kernel.cc index 978bdb5129c..c5fca8881e2 100644 --- a/paddle/phi/kernels/xpu/full_kernel.cc +++ b/paddle/phi/kernels/xpu/full_kernel.cc @@ -70,7 +70,7 @@ void FullLikeKernel(const Context& dev_ctx, DataType dtype, DenseTensor* out) { dev_ctx.template Alloc(out); - auto value = val.to(); + auto value = val.to(); using XPUInTDType = typename XPUTypeTrait::Type; using CommonType = typename std::common_type< float, -- GitLab