fix fused linear bfloat16 (#51384)

3328a3d5 · sneaxiy · GitHub · 2c7cbb86 · 3328a3d5
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu +2 -1

未找到文件。
--- 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