Skip to content

Commit 8c5e841

Browse files
authored
Fix bugs for pir current_expected_place in paddle/device (#67892)
1 parent 5c7df1e commit 8c5e841

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

python/paddle/device/__init__.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -500,7 +500,7 @@ def __init__(
500500
interprocess: bool = False,
501501
) -> None:
502502
if device is None:
503-
self.device = paddle.framework._current_expected_place()
503+
self.device = paddle.framework._current_expected_place_()
504504
elif isinstance(device, str):
505505
self.device = paddle.device._convert_to_place(device)
506506
else:
@@ -686,7 +686,7 @@ def __init__(
686686
return
687687

688688
if device is None:
689-
self.device = paddle.framework._current_expected_place()
689+
self.device = paddle.framework._current_expected_place_()
690690
elif isinstance(device, str):
691691
self.device = paddle.device._convert_to_place(device)
692692
else:
@@ -885,7 +885,7 @@ def current_stream(device: PlaceLike | None = None) -> Stream:
885885
886886
'''
887887
if device is None:
888-
place = paddle.framework._current_expected_place()
888+
place = paddle.framework._current_expected_place_()
889889
elif isinstance(device, str):
890890
place = paddle.device._convert_to_place(device)
891891
else:
@@ -996,7 +996,7 @@ def __enter__(self) -> None:
996996

997997
self.src_prev_stream = current_stream(cur_stream.device)
998998
if self.src_prev_stream.device != cur_stream.device:
999-
self.tmp_place = paddle.base.framework._current_expected_place()
999+
self.tmp_place = paddle.base.framework._current_expected_place_()
10001000
paddle.base.framework._set_expected_place(cur_stream.device)
10011001
self.dst_prev_stream = current_stream(cur_stream.device)
10021002
set_stream(cur_stream)
@@ -1046,7 +1046,7 @@ def synchronize(device: PlaceLike | None = None) -> None:
10461046
"""
10471047

10481048
if device is None:
1049-
place = paddle.framework._current_expected_place()
1049+
place = paddle.framework._current_expected_place_()
10501050
elif isinstance(device, str):
10511051
place = paddle.device._convert_to_place(device)
10521052
else:

0 commit comments

Comments
 (0)