From 7d000112e2263a5dcd40182074a55ebe9e4f0442 Mon Sep 17 00:00:00 2001
From: sneaxiy <32832641+sneaxiy@users.noreply.github.com>
Date: Fri, 9 Sep 2022 09:39:54 +0800
Subject: [PATCH] fix fused_gemm_epilogue compile error (#45899)

---
 paddle/fluid/operators/fused/fused_gemm_epilogue_op.h | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.h b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.h
index 62701fe43f..0f4fa9f889 100644
--- a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.h
+++ b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.h
@@ -15,8 +15,12 @@ limitations under the License. */
 
 #pragma once
 
-#if defined(PADDLE_WITH_CUDA) && CUDA_VERSION >= 11060
+#ifdef PADDLE_WITH_CUDA
+
 #include <cuda_runtime_api.h>
+#include "cuda.h"  // NOLINT
+
+#if CUDA_VERSION >= 11060
 
 #include <algorithm>
 #include <mutex>
@@ -322,4 +326,6 @@ class GemmEpilogueAlgoCache {
 
 }  // namespace operators
 }  // namespace paddle
+
+#endif
 #endif
-- 
GitLab