Merge pull request #3097 from QiJune/fix_bug_memory

fix bug in memory GetGPUBuddyAllocator
8 years ago · 44486b6f97
parent c54f14d89d bd8d2d8ba3
commit 44486b6f97
1 changed files with 1 additions and 0 deletions
--- a/paddle/memory/memory.cc
+++ b/paddle/memory/memory.cc
@ -60,6 +60,7 @@ detail::BuddyAllocator* GetGPUBuddyAllocator(int gpu_id) {
                                           platform::GpuMaxChunkSize());
    }
  }
+  platform::SetDeviceId(gpu_id);
  return as[gpu_id];
 }