@@ -500,7 +500,7 @@ def __init__(
500
500
interprocess : bool = False ,
501
501
) -> None :
502
502
if device is None :
503
- self .device = paddle .framework ._current_expected_place ()
503
+ self .device = paddle .framework ._current_expected_place_ ()
504
504
elif isinstance (device , str ):
505
505
self .device = paddle .device ._convert_to_place (device )
506
506
else :
@@ -686,7 +686,7 @@ def __init__(
686
686
return
687
687
688
688
if device is None :
689
- self .device = paddle .framework ._current_expected_place ()
689
+ self .device = paddle .framework ._current_expected_place_ ()
690
690
elif isinstance (device , str ):
691
691
self .device = paddle .device ._convert_to_place (device )
692
692
else :
@@ -885,7 +885,7 @@ def current_stream(device: PlaceLike | None = None) -> Stream:
885
885
886
886
'''
887
887
if device is None :
888
- place = paddle .framework ._current_expected_place ()
888
+ place = paddle .framework ._current_expected_place_ ()
889
889
elif isinstance (device , str ):
890
890
place = paddle .device ._convert_to_place (device )
891
891
else :
@@ -996,7 +996,7 @@ def __enter__(self) -> None:
996
996
997
997
self .src_prev_stream = current_stream (cur_stream .device )
998
998
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_ ()
1000
1000
paddle .base .framework ._set_expected_place (cur_stream .device )
1001
1001
self .dst_prev_stream = current_stream (cur_stream .device )
1002
1002
set_stream (cur_stream )
@@ -1046,7 +1046,7 @@ def synchronize(device: PlaceLike | None = None) -> None:
1046
1046
"""
1047
1047
1048
1048
if device is None :
1049
- place = paddle .framework ._current_expected_place ()
1049
+ place = paddle .framework ._current_expected_place_ ()
1050
1050
elif isinstance (device , str ):
1051
1051
place = paddle .device ._convert_to_place (device )
1052
1052
else :
0 commit comments