From 3328a3d5d244169b5928244cde75c7232313c0be Mon Sep 17 00:00:00 2001
From: sneaxiy <32832641+sneaxiy@users.noreply.github.com>
Date: Thu, 9 Mar 2023 08:54:45 +0800
Subject: [PATCH] fix fused linear bfloat16 (#51384)

---
 paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu
index a6a136391de..6e9bd009a7d 100644
--- a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu
+++ b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu
@@ -130,5 +130,6 @@ REGISTER_OP_CUDA_KERNEL(
     ops::FusedGemmEpilogueGradKernel<phi::GPUContext, double>,
     ops::FusedGemmEpilogueGradKernel<phi::GPUContext,
                                      paddle::platform::float16>,
-    ops::FusedGemmEpilogueKernel<phi::GPUContext, paddle::platform::bfloat16>);
+    ops::FusedGemmEpilogueGradKernel<phi::GPUContext,
+                                     paddle::platform::bfloat16>);
 #endif
-- 
GitLab