fix bug in memory GetGPUBuddyAllocator

8 years ago · bd8d2d8ba3
parent c54f14d89d
commit bd8d2d8ba3
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];
 }