From e43a527ba64fe18e12949e086ac569b097ab18a0 Mon Sep 17 00:00:00 2001 From: Aurelius84 Date: Tue, 7 Mar 2023 16:38:12 +0800 Subject: [PATCH] Fix fused_gemm_epilogue compilation not declared problem (#51280) --- paddle/fluid/operators/fused/fused_gemm_epilogue_op.h | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.h b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.h index 059cf66fac6..f3b21b144cd 100644 --- a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.h +++ b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.h @@ -28,6 +28,7 @@ limitations under the License. */ #include "gflags/gflags.h" #include "paddle/fluid/framework/scope_guard.h" +#include "paddle/fluid/memory/memory.h" #include "paddle/fluid/platform/dynload/cublasLt.h" #include "paddle/fluid/platform/enforce.h" #include "paddle/fluid/platform/float16.h" -- GitLab