diff --git a/paddle/fluid/platform/event.h b/paddle/fluid/platform/event.h index a4db23758b1c477114cd03dcd0e9f51296c575c6..2dcf966754cbed2670acb9c3548c23355be5503c 100644 --- a/paddle/fluid/platform/event.h +++ b/paddle/fluid/platform/event.h @@ -14,6 +14,9 @@ limitations under the License. */ #pragma once #include +#ifdef PADDLE_WITH_CUDA +#include +#endif namespace paddle { namespace platform {