From 3e9aa7fd8bfac7434057afcdd6ae62ea7a92bff1 Mon Sep 17 00:00:00 2001
From: liaogang <liaogang@baidu.com>
Date: Wed, 28 Jun 2017 10:42:54 +0800
Subject: [PATCH] FIX: Pass CI

---
 cmake/generic.cmake                     | 4 ----
 paddle/memory/CMakeLists.txt            | 6 ------
 paddle/memory/detail/buddy_allocator.cc | 2 +-
 paddle/memory/detail/buddy_allocator.h  | 3 +++
 4 files changed, 4 insertions(+), 11 deletions(-)

diff --git a/cmake/generic.cmake b/cmake/generic.cmake
index 840155750e1..69e8164a00d 100644
--- a/cmake/generic.cmake
+++ b/cmake/generic.cmake
@@ -78,10 +78,6 @@
 #
 #   cc_test(example_test SRCS example_test.cc DEPS example glog gflags)
 
-if(WITH_GPU)
-  add_definitions(-DPADDLE_WITH_GPU)
-endif()
-
 if(NOT APPLE)
     find_package(Threads REQUIRED)
     link_libraries(${CMAKE_THREAD_LIBS_INIT})
diff --git a/paddle/memory/CMakeLists.txt b/paddle/memory/CMakeLists.txt
index e74ce75c939..3943c3cfad3 100644
--- a/paddle/memory/CMakeLists.txt
+++ b/paddle/memory/CMakeLists.txt
@@ -1,7 +1 @@
 add_subdirectory(detail)
-
-if(${WITH_GPU})
-  nv_library(memory SRCS memory.cc)
-else(${WITH_GPU})
-  cc_library(memory SRCS memory.cc)
-endif(${WITH_GPU})
diff --git a/paddle/memory/detail/buddy_allocator.cc b/paddle/memory/detail/buddy_allocator.cc
index 895bf319d77..ebe680f5eea 100644
--- a/paddle/memory/detail/buddy_allocator.cc
+++ b/paddle/memory/detail/buddy_allocator.cc
@@ -14,7 +14,7 @@
 
 #pragma once
 
-#include "paddle/memory/detail/system_allocator.h"
+#include "paddle/memory/detail/buddy_allocator.h"
 
 namespace paddle {
 namespace memory {
diff --git a/paddle/memory/detail/buddy_allocator.h b/paddle/memory/detail/buddy_allocator.h
index 129b137ed73..702c7d28ee5 100644
--- a/paddle/memory/detail/buddy_allocator.h
+++ b/paddle/memory/detail/buddy_allocator.h
@@ -16,6 +16,9 @@
 
 #include "paddle/memory/detail/system_allocator.h"
 
+#include <vector>
+#include <mutex>
+
 namespace paddle {
 namespace memory {
 namespace detail {
-- 
GitLab