fix build issue for cudaEvent_t

test=develop
6 years ago · b5d6e38b05
parent 4bd28b304b
commit b5d6e38b05
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 "paddle/fluid/platform/gpu_info.h"
+#endif

 namespace paddle {
 namespace platform {