[CustomDevice] remove get_device in device_guard (#57994)

ronny1996 · web-flow · commit c24ee9eb2a94 · 2023-10-11T10:53:17.000+08:00
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_;
 };
 
