fix no place.device

PaddlePaddle · Apr 5, 2022 · a0835a1 · a0835a1
1 parent 8be37f5
commit a0835a1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/paddle/fluid/imperative/tracer.cc b/paddle/fluid/imperative/tracer.cc
@@ -387,7 +387,9 @@ void Tracer::SetExpectedPlace(platform::Place place) {
   expected_place_ = place;
   if (platform::is_gpu_place(place)) {
 #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
-    platform::SetDeviceId(place.device);
+    if (place.device) {
+      platform::SetDeviceId(place.device);
+    }
 #else
     PADDLE_THROW(platform::errors::PreconditionNotMet(
         "PaddlePaddle should compile with GPU if use CUDAPlace."));