提交 176277b8 编写于 作者: Y Yu Yang

Add log

上级 ed7727e8
...@@ -90,6 +90,7 @@ size_t Used<platform::CUDAPlace>(platform::CUDAPlace place) { ...@@ -90,6 +90,7 @@ size_t Used<platform::CUDAPlace>(platform::CUDAPlace place) {
template <> template <>
void* Alloc<platform::CUDAPlace>(platform::CUDAPlace place, size_t size) { void* Alloc<platform::CUDAPlace>(platform::CUDAPlace place, size_t size) {
auto* buddy_allocator = GetGPUBuddyAllocator(place.device); auto* buddy_allocator = GetGPUBuddyAllocator(place.device);
VLOG(30) << "Allocating " << size << " bytes on " << place;
auto* ptr = buddy_allocator->Alloc(size); auto* ptr = buddy_allocator->Alloc(size);
if (ptr == nullptr) { if (ptr == nullptr) {
int cur_dev = platform::GetCurrentDeviceId(); int cur_dev = platform::GetCurrentDeviceId();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册