Skip to content

Commit 7753f2b

Browse files
committed
[API Compatibility] Fix _device_to_paddle (#77490)
* fix python/paddle/device/__init__.py * unify use device_to_place
1 parent 34c144a commit 7753f2b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/paddle/device/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -871,7 +871,7 @@ def get_device_properties(
871871
>>> # paddle.device.get_device_properties('npu')
872872
>>> # _customDeviceProperties(name='', major=0, minor=0, total_memory=0MB, multi_processor_count=0)
873873
"""
874-
device = _device_to_paddle(device)
874+
device = device_to_place(device)
875875
return _get_device_properties(device)
876876

877877

@@ -1513,8 +1513,8 @@ def _device_to_paddle(
15131513
elif isinstance(dev, str):
15141514
cleaned_device = dev.strip()
15151515
return (
1516-
cleaned_device.replace("cuda:", "gpu:")
1517-
if "cuda:" in cleaned_device
1516+
cleaned_device.replace("cuda", "gpu")
1517+
if "cuda" in cleaned_device
15181518
else cleaned_device
15191519
)
15201520
elif dev is None:

0 commit comments

Comments
 (0)