Fix fused_gemm_epilogue compilation not declared problem (#51280)

e43a527b · Aurelius84 · GitHub · b5232bf4 · e43a527b
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/operators/fused/fused_gemm_epilogue_op.h paddle/fluid/operators/fused/fused_gemm_epilogue_op.h +1 -0

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