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

fix fused_gemm_epilogue compile error (#45899)

上级 5d7e1c91
......@@ -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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册