From 6e7209f0584f73eb22313d98c676333379736d1e Mon Sep 17 00:00:00 2001
From: liaogang <liaogang@baidu.com>
Date: Thu, 29 Jun 2017 16:30:32 +0800
Subject: [PATCH] ENH: Add gpu info interface

---
 paddle/platform/gpu_info.cc | 25 +++++++++++++++++++++++++
 paddle/platform/gpu_info.h  |  6 ++++++
 2 files changed, 31 insertions(+)

diff --git a/paddle/platform/gpu_info.cc b/paddle/platform/gpu_info.cc
index 4208d8307..d6c6fe34e 100644
--- a/paddle/platform/gpu_info.cc
+++ b/paddle/platform/gpu_info.cc
@@ -42,8 +42,33 @@ size_t GpuMaxAllocSize() {
 
   GpuMemoryUsage(available, total);
 
+  // Reserve the rest for page tables, etc.
   return total * FLAGS_fraction_of_gpu_memory_to_use;
 }
 
+size_t GpuMinChunkSize() {
+  // Allow to allocate the minimum chunk size is 256 bytes.
+  return 1 << 8;
+}
+
+size_t GpuMaxChunkSize() {
+  // Allow to allocate the maximum chunk size is roughly 3% of CPU memory.
+  size_t total = 0;
+  size_t available = 0;
+
+  GpuMemoryUsage(available, total);
+
+  // Reserving the rest memory for page tables, etc.
+  size_t reserving = (1 - FLAGS_fraction_of_gpu_memory_to_use) * total;
+
+  // If available less than minimum chunk size, no usable memory exists.
+  available = std::max(available, GpuMinChunkSize()) - GpuMinChunkSize();
+
+  // If available less than reserving, no usable memory exists.
+  size_t usable = std::max(available, reserving) - reserving;
+
+  return usable;
+}
+
 }  // namespace platform
 }  // namespace paddle
diff --git a/paddle/platform/gpu_info.h b/paddle/platform/gpu_info.h
index 174f093b4..d7bf0f409 100644
--- a/paddle/platform/gpu_info.h
+++ b/paddle/platform/gpu_info.h
@@ -30,6 +30,12 @@ void GpuMemoryUsage(size_t& available, size_t& total);
 //! Get the maximum allocation size of current GPU device.
 size_t GpuMaxAllocSize();
 
+//! Get the minimum chunk size for GPU buddy allocator.
+size_t GpuMinChunkSize();
+
+//! Get the maximum chunk size for GPU buddy allocator.
+size_t GpuMaxChunkSize();
+
 }  // namespace platform
 }  // namespace paddle
 
-- 
GitLab