diff --git a/paddle/memory/memory_test.cc b/paddle/memory/memory_test.cc
index 458c8b2e24fa29c010d01cb4a4e70f1b46bcda27..e13cbabb268a8e51c0c173a010694c2ebb193294 100644
--- a/paddle/memory/memory_test.cc
+++ b/paddle/memory/memory_test.cc
@@ -34,13 +34,6 @@ size_t align(size_t size, paddle::platform::CPUPlace place) {
   return remaining == 0 ? size : size + (alignment - remaining);
 }
 
-size_t align(size_t size, paddle::platform::GPUPlace place) {
-  size += sizeof(paddle::memory::detail::Metadata);
-  size_t alignment = paddle::platform::GpuMinChunkSize();
-  size_t remaining = size % alignment;
-  return remaining == 0 ? size : size + (alignment - remaining);
-}
-
 void update_size(size_t &total_size, const size_t size) {}
 
 TEST(BuddyAllocator, CPUAllocation) {
@@ -91,6 +84,13 @@ TEST(BuddyAllocator, CPUMultAlloc) {
 
 #ifndef PADDLE_ONLY_CPU
 
+size_t align(size_t size, paddle::platform::GPUPlace place) {
+  size += sizeof(paddle::memory::detail::Metadata);
+  size_t alignment = paddle::platform::GpuMinChunkSize();
+  size_t remaining = size % alignment;
+  return remaining == 0 ? size : size + (alignment - remaining);
+}
+
 TEST(BuddyAllocator, GPUAllocation) {
   void *p = nullptr;