Merge pull request #15916 from wopeizl/win/fixevent1

fix build issue for cudaEvent_t
6 years ago · 2192c46436
parent 660e410655 c6472579c0
commit 2192c46436
1 changed files with 3 additions and 0 deletions
--- a/paddle/fluid/platform/event.h
+++ b/paddle/fluid/platform/event.h
@ -14,6 +14,9 @@ limitations under the License. */

 #pragma once
 #include <string>
+#ifdef PADDLE_WITH_CUDA
+#include <cuda_runtime.h>
+#endif

 namespace paddle {
 namespace platform {