提交 ca9d21f5 编写于 作者: D Dun Liang

Fix #12578: Wrong error message when run out of GPU memory

上级 e658762a
...@@ -162,6 +162,8 @@ void BuddyAllocator::Free(void* p) { ...@@ -162,6 +162,8 @@ void BuddyAllocator::Free(void* p) {
} }
size_t BuddyAllocator::Used() { return total_used_; } size_t BuddyAllocator::Used() { return total_used_; }
size_t BuddyAllocator::GetMinChunkSize() {return min_chunk_size_;};
size_t BuddyAllocator::GetMaxChunkSize() {return max_chunk_size_;};
void* BuddyAllocator::SystemAlloc(size_t size) { void* BuddyAllocator::SystemAlloc(size_t size) {
size_t index = 0; size_t index = 0;
......
...@@ -42,6 +42,8 @@ class BuddyAllocator { ...@@ -42,6 +42,8 @@ class BuddyAllocator {
void* Alloc(size_t unaligned_size); void* Alloc(size_t unaligned_size);
void Free(void* ptr); void Free(void* ptr);
size_t Used(); size_t Used();
size_t GetMinChunkSize();
size_t GetMaxChunkSize();
public: public:
// Disable copy and assignment // Disable copy and assignment
......
...@@ -119,8 +119,8 @@ void* Alloc<platform::CUDAPlace>(platform::CUDAPlace place, size_t size) { ...@@ -119,8 +119,8 @@ void* Alloc<platform::CUDAPlace>(platform::CUDAPlace place, size_t size) {
LOG(WARNING) << "Cannot allocate " << size << " bytes in GPU " LOG(WARNING) << "Cannot allocate " << size << " bytes in GPU "
<< place.device << ", available " << avail << " bytes"; << place.device << ", available " << avail << " bytes";
LOG(WARNING) << "total " << total; LOG(WARNING) << "total " << total;
LOG(WARNING) << "GpuMinChunkSize " << platform::GpuMinChunkSize(); LOG(WARNING) << "GpuMinChunkSize " << buddy_allocator->GetMinChunkSize();
LOG(WARNING) << "GpuMaxChunkSize " << platform::GpuMaxChunkSize(); LOG(WARNING) << "GpuMaxChunkSize " << buddy_allocator->GetMaxChunkSize();
LOG(WARNING) << "GPU memory used: " << Used<platform::CUDAPlace>(place); LOG(WARNING) << "GPU memory used: " << Used<platform::CUDAPlace>(place);
platform::SetDeviceId(cur_dev); platform::SetDeviceId(cur_dev);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册