fix sumraw (#72138)

aquagull · web-flow · commit fe383c810d7d · 2025-04-17T10:46:51.000+08:00
* fix sumraw

* add test

* empty

* fix
diff --git a/paddle/phi/kernels/cpu/reduce_sum_kernel.cc b/paddle/phi/kernels/cpu/reduce_sum_kernel.cc
@@ -86,8 +86,7 @@ void SumRawKernel(const Context& dev_ctx,
     }
     out->Resize(phi::make_ddim(out_dims));
     dev_ctx.template Alloc<T>(out);
-    FullKernel<T, Context>(
-        dev_ctx, out_dims, 0, phi::CppTypeToDataType<T>::Type(), out);
+    FullKernel<T, Context>(dev_ctx, out_dims, 0, out_dtype, out);
     return;
   }
   phi::Reduce<CPUContext, T, phi::funcs::SumFunctor>(
diff --git a/paddle/phi/kernels/kps/reduce_kernel.cu b/paddle/phi/kernels/kps/reduce_kernel.cu
@@ -255,16 +255,8 @@ void SumRawKernel(const Context& dev_ctx,
       }
     }
     out->Resize(phi::make_ddim(out_dims));
-    if (x.dtype() == phi::DataType::BOOL || x.dtype() == phi::DataType::INT32) {
-      dev_ctx.template Alloc<int64_t>(out);
-      FullKernel<int64_t, Context>(
-          dev_ctx, out_dims, 0, phi::CppTypeToDataType<int64_t>::Type(), out);
-    } else {
-      dev_ctx.template Alloc<T>(out);
-      FullKernel<T, Context>(
-          dev_ctx, out_dims, 0, phi::CppTypeToDataType<T>::Type(), out);
-    }
-
+    dev_ctx.template Alloc<T>(out);
+    FullKernel<T, Context>(dev_ctx, out_dims, 0, out_dtype, out);
     return;
   }
   if (x.numel() > std::numeric_limits<int32_t>::max()) {
diff --git a/test/legacy_test/test_sum_op.py b/test/legacy_test/test_sum_op.py
@@ -569,6 +569,35 @@ def test_dtype1():
             self.assertRaises(TypeError, test_dtype1)
 
 
+class TestSumOpDtype(unittest.TestCase):
+    def setUp(self):
+        self.shape = [0, 1, 1]
+        self.axis = 0
+        self.input_dtype = 'int32'
+        self.output_dtype = 'int32'
+        self.paddle_output_dtype = paddle.int32
+
+    def test_dygraph(self):
+        with dygraph_guard():
+            x_paddle = paddle.zeros(shape=self.shape, dtype=self.input_dtype)
+            paddle_result = x_paddle.sum(
+                axis=self.axis, dtype=self.output_dtype
+            )
+
+        self.assertEqual(paddle_result.dtype, self.paddle_output_dtype)
+
+    def test_static(self):
+        with static_guard():
+            with paddle.static.program_guard(
+                paddle.static.Program(), paddle.static.Program()
+            ):
+                x = paddle.static.data(
+                    name='x', shape=self.shape, dtype=self.input_dtype
+                )
+                result = paddle.sum(x, axis=self.axis, dtype=self.output_dtype)
+                self.assertEqual(result[0].dtype, self.paddle_output_dtype)
+
+
 class TestSumOpError(unittest.TestCase):
     def test_errors(self):
         def test_empty_list_input():

Original file line number	Diff line number	Diff line change
`@@ -86,8 +86,7 @@ void SumRawKernel(const Context& dev_ctx,`
`86`	`86`	`}`
`87`	`87`	`out->Resize(phi::make_ddim(out_dims));`
`88`	`88`	`dev_ctx.template Alloc<T>(out);`
`89`		`- FullKernel<T, Context>(`
`90`		`- dev_ctx, out_dims, 0, phi::CppTypeToDataType<T>::Type(), out);`
	`89`	`+ FullKernel<T, Context>(dev_ctx, out_dims, 0, out_dtype, out);`
`91`	`90`	`return;`
`92`	`91`	`}`
`93`	`92`	`phi::Reduce<CPUContext, T, phi::funcs::SumFunctor>(`
Original file line number	Diff line number	Diff line change
`@@ -255,16 +255,8 @@ void SumRawKernel(const Context& dev_ctx,`
`255`	`255`	`}`
`256`	`256`	`}`
`257`	`257`	`out->Resize(phi::make_ddim(out_dims));`
`258`		`- if (x.dtype() == phi::DataType::BOOL \|\| x.dtype() == phi::DataType::INT32) {`
`259`		`- dev_ctx.template Alloc<int64_t>(out);`
`260`		`- FullKernel<int64_t, Context>(`
`261`		`- dev_ctx, out_dims, 0, phi::CppTypeToDataType<int64_t>::Type(), out);`
`262`		`- } else {`
`263`		`- dev_ctx.template Alloc<T>(out);`
`264`		`- FullKernel<T, Context>(`
`265`		`- dev_ctx, out_dims, 0, phi::CppTypeToDataType<T>::Type(), out);`
`266`		`- }`
`267`		`-`
	`258`	`+ dev_ctx.template Alloc<T>(out);`
	`259`	`+ FullKernel<T, Context>(dev_ctx, out_dims, 0, out_dtype, out);`
`268`	`260`	`return;`
`269`	`261`	`}`
`270`	`262`	`if (x.numel() > std::numeric_limits<int32_t>::max()) {`