@@ -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