fix current expected place error in multicard (#43664)

abbd4abb · Jiabin Yang · GitHub · b6bc6f7a · abbd4abb
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

python/paddle/fluid/framework.py python/paddle/fluid/framework.py +3 -3

未找到文件。
--- a/python/paddle/fluid/framework.py
+++ b/python/paddle/fluid/framework.py
@@ -574,7 +574,7 @@ def _current_expected_place():
            except Exception as e:
                device_count = 0
            if device_count > 0:
-                _global_expected_place_ = core.CUDAPlace(0)
+                _global_expected_place_ = core.CUDAPlace(_cuda_ids()[0])
            else:
                warnings.warn(
                    "You are using GPU version Paddle, but your CUDA device is not set properly. CPU device will be used by default."
@@ -586,7 +586,7 @@ def _current_expected_place():
            except Exception as e:
                device_count = 0
            if device_count > 0:
-                _global_expected_place_ = core.XPUPlace(0)
+                _global_expected_place_ = core.XPUPlace(_xpu_ids()[0])
            else:
                warnings.warn(
                    "You are using XPU version Paddle, but your XPU device is not set properly. CPU device will be used by default."
@@ -598,7 +598,7 @@ def _current_expected_place():
            except Exception as e:
                device_count = 0
            if device_count > 0:
-                _global_expected_place_ = core.MLUPlace(0)
+                _global_expected_place_ = core.MLUPlace(_mlu_ids()[0])
            else:
                warnings.warn(
                    "You are using MLU version Paddle, but your MLU device is not set properly. CPU device will be used by default."