[CustomDevice] remove get_device in device_guard (PaddlePaddle#57994)

danleifeng · Oct 11, 2023 · 85b5598 · 85b5598
1 parent 0596610
commit 85b5598
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/paddle/phi/backends/device_guard.h b/paddle/phi/backends/device_guard.h
@@ -20,20 +20,15 @@ namespace phi {
 class DeviceGuard {
  public:
   explicit inline DeviceGuard(const Place& place)
-      : dev_type_(place.GetDeviceType()) {
-    prev_id = DeviceManager::GetDevice(dev_type_);
-    cur_id = place.GetDeviceId();
-
-    if (cur_id != prev_id) {
-      DeviceManager::SetDevice(dev_type_, cur_id);
-    }
+      : cur_id(place.GetDeviceId()), dev_type_(place.GetDeviceType()) {
+    DeviceManager::SetDevice(dev_type_, cur_id);
   }
 
   DeviceGuard(const DeviceGuard& o) = delete;
   DeviceGuard& operator=(const DeviceGuard& o) = delete;
 
  private:
-  size_t prev_id, cur_id;
+  size_t cur_id;
   std::string dev_type_;
 };