Register bf16 type for c_reduce_sum op (#61353)

From00 · web-flow · commit 02575fb13e3b · 2024-01-31T10:53:25.000+08:00
diff --git a/paddle/fluid/operators/collective/c_reduce_sum_op.cu.cc b/paddle/fluid/operators/collective/c_reduce_sum_op.cu.cc
@@ -31,4 +31,5 @@ PD_REGISTER_STRUCT_KERNEL(c_reduce_sum,
                           double,
                           int,
                           int64_t,
-                          plat::float16) {}
+                          plat::float16,
+                          plat::bfloat16) {}
