From ec8115866c2713158532e8eb6755a79d5052563c Mon Sep 17 00:00:00 2001
From: runzhech <enzo9533@hotmail.com>
Date: Thu, 11 Apr 2024 03:57:21 +0000
Subject: [PATCH] support bf16 c_concat

---
 paddle/fluid/operators/collective/c_concat_op_xpu.cc | 3 ++-
 paddle/phi/backends/xpu/xpu3_op_list.cc              | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/collective/c_concat_op_xpu.cc b/paddle/fluid/operators/collective/c_concat_op_xpu.cc
index 10a2624ae83a4e..8e9fc8c4f988b1 100644
--- a/paddle/fluid/operators/collective/c_concat_op_xpu.cc
+++ b/paddle/fluid/operators/collective/c_concat_op_xpu.cc
@@ -169,4 +169,5 @@ PD_REGISTER_STRUCT_KERNEL(c_concat,
                           float,
                           int,
                           int64_t,
-                          plat::float16) {}
+                          plat::float16,
+                          plat::bfloat16) {}
diff --git a/paddle/phi/backends/xpu/xpu3_op_list.cc b/paddle/phi/backends/xpu/xpu3_op_list.cc
index 48dc5d83341931..690b3be25fb4ee 100644
--- a/paddle/phi/backends/xpu/xpu3_op_list.cc
+++ b/paddle/phi/backends/xpu/xpu3_op_list.cc
@@ -125,6 +125,7 @@ XPUOpMap& get_kl3_ops() {
                      phi::DataType::INT64})},
       {"c_concat",
        XPUKernelSet({phi::DataType::FLOAT16,
+                     phi::DataType::BFLOAT16,
                      phi::DataType::FLOAT32,
                      phi::DataType::INT32,
                      phi::DataType::INT64})},