diff --git a/paddle/fluid/imperative/nccl_context.h b/paddle/fluid/imperative/nccl_context.h index 51c86190439d1739a99bc91a712f663383bb9371..b4f44e56405a51082e60afd69fb6f011dab44b86 100644 --- a/paddle/fluid/imperative/nccl_context.h +++ b/paddle/fluid/imperative/nccl_context.h @@ -14,7 +14,7 @@ #pragma once // network header files -#ifndef _WIN32 +#if defined(PADDLE_WITH_CUDA) && !defined(_WIN32) #include #include #include @@ -26,7 +26,7 @@ #include "paddle/fluid/framework/variable.h" #include "paddle/fluid/platform/device_context.h" -#ifdef PADDLE_WITH_CUDA +#if defined(PADDLE_WITH_CUDA) && !defined(_WIN32) #include "paddle/fluid/platform/dynload/nccl.h" #endif #include "paddle/fluid/platform/place.h"