diff --git a/paddle/fluid/platform/event.h b/paddle/fluid/platform/event.h index a4db23758b1c477114cd03dcd0e9f51296c575c6..5e52ccfbfbb262d4e691dab3e692fb981b2f97d1 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 "paddle/fluid/platform/gpu_info.h" +#endif namespace paddle { namespace platform {