zjUddlmZddlmZmZddlmZddlmZerddl m Z ddl m Z ee e efZded<d gZ dddZd S)) annotations) TYPE_CHECKINGUnion) TypeAlias)core) CustomPlace)_customDevicePropertiesr_CustomPlaceLikeget_device_propertiesNdevice_CustomPlaceLike | Nonereturnr c|t|tr0|}tj}|st d|d}nt|tjr)|}|}nt|trn| d}|dkr|}d}n|d|}||dzd}| st d|dt|}nAt d |d tj}|st d|d}d}tj ||S) a Return the properties of given custom device. Args: device (CustomPlace|str|int|None, optional): The device, the id of the device or the string name of device like 'metax_gpu:x' which to get the properties of the device from. Notice that this api only supports gpgpu backend. If device is None, the device is the current device. Default: None. Returns: _customDeviceProperties: The properties of the device which include device name, major compute capability, minor compute capability, global memory available and the number of multiprocessors on the device. Examples: .. code-block:: python >>> import paddle >>> paddle.device.get_device_properties('metax_gpu:0') >>> paddle.device.get_device_properties(0) >>> paddle.device.get_device_properties(paddle.CustomPlace('metax_gpu', 0)) Nz No custom device types availabler:zInvalid device ID 'z:'. After colon must be digits only. Example: 'metax_gpu:0'zThe device type z is not expected. Because paddle.device.get_device_properties only support int, str or CustomPlace. Please input appropriate device again!) isinstanceintrget_all_custom_device_type ValueErrorrget_device_type get_device_idstrrfindisdigitr )r device_id dev_types device_name colon_idx device_id_strs k/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/device/gpgpu_backend.pyr r %s2 fc " " I799I E !CDDD#A,KK  0 1 1  0022K,,..II  $ $  S))IB$  $ZiZ0 &y1} 7 $,,..$1m111  .. 96999 355  A?@@ @l   %k9 = ==)N)r r rr ) __future__rtypingrrtyping_extensionsr paddle.baserpaddlerpaddle.base.libpaddler rrr __annotations____all__r r"r!r,s#""""""''''''''''''''""""""======"'   #  '+C>C>C>C>C>C>C>r"