From 169829c83a5b893f21c52454c3728865abefc0b9 Mon Sep 17 00:00:00 2001 From: Yan Xu Date: Sun, 7 Apr 2019 02:05:56 +0800 Subject: [PATCH] fix win gpu test=develop (#16694) --- paddle/fluid/imperative/nccl_context.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/imperative/nccl_context.h b/paddle/fluid/imperative/nccl_context.h index 51c86190439..b4f44e56405 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" -- GitLab