FIX: Pass CI

8 years ago · 3e9aa7fd8b
parent 55648b4f4c
commit 3e9aa7fd8b
4 changed files with 4 additions and 11 deletions
--- 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})
--- 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})
--- 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 {
--- 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 {