未验证 提交 7d000112 编写于 作者: S sneaxiy 提交者: GitHub

fix fused_gemm_epilogue compile error (#45899)

上级 5d7e1c91
...@@ -15,8 +15,12 @@ limitations under the License. */ ...@@ -15,8 +15,12 @@ limitations under the License. */
#pragma once #pragma once
#if defined(PADDLE_WITH_CUDA) && CUDA_VERSION >= 11060 #ifdef PADDLE_WITH_CUDA
#include <cuda_runtime_api.h> #include <cuda_runtime_api.h>
#include "cuda.h" // NOLINT
#if CUDA_VERSION >= 11060
#include <algorithm> #include <algorithm>
#include <mutex> #include <mutex>
...@@ -322,4 +326,6 @@ class GemmEpilogueAlgoCache { ...@@ -322,4 +326,6 @@ class GemmEpilogueAlgoCache {
} // namespace operators } // namespace operators
} // namespace paddle } // namespace paddle
#endif
#endif #endif
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册