Αi O%SSKJr SSKJrJr SSKJr SSKrSSKJ r SSK J r J r J r Jr \(aSSKJr \\\\4rS \S '\ R("5r\(a\SOSr\(a\ R."\5(d\"S 5e\(a \S ;aSS KJr OSS KJr /SQrSSSjjrSSjrSSSjjrSSSjjr SS Sjjr!SS Sjjr"SSSjjr#SSSjjr$SS!Sjjr%SS Sjjr&SS"Sjjr'SS#Sjjr(S$Sjr)g)%) annotations) TYPE_CHECKINGUnion) TypeAliasN)core)EventStream create_event create_stream) CustomPlacer_CustomPlaceLikezLNo custom device available, please install paddle with custom device support) metax_gpu iluvatar_gpu)get_device_properties) r r device_countr empty_cachemax_memory_allocatedmax_memory_reservedreset_max_memory_allocatedreset_max_memory_reservedmemory_allocatedmemory_reservedcurrent_stream synchronizecxU(a[R"U5nU$[R"[5nU$)a Return the number of custom devices available. Args: device_type (str, optional): The type of custom device (e.g., 'npu', 'mlu', etc.). If None, returns the count of the first available custom device type. Returns: int: the number of custom devices available. Examples: .. code-block:: python >>> import paddle >>> paddle.device.device_count() >>> paddle.device.device_count('npu') )rget_custom_device_countdev_type) device_typenums [/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/device/custom_device.pyrrEs5&**;7 J**84 Jc.[R"5 g)z Releases idle cached memory held by the allocator so that those can be used in other GPU application and visible in device-specific tools. Examples: .. code-block:: python >>> import paddle >>> paddle.device.empty_cache() N)rdevice_empty_cacher"r!rr`s r"cSnUcSnO[U[5(aQURS5nUS:XaSnOXS-SnUR5(d[ SUS35e[ U5nOW[U[ 5(aUnO?[U[ R5(aUR5nO[ SUS 35e[ R"S U5$) a Return the peak size of memory that is allocated to tensor of the given device. Args: device(_CustomPlaceLike, optional): Support input like 'npu:0', 'mlu', int, or CustomPlace. If None, the device is the first available custom device with index 0. Returns: int: The peak size of memory that is allocated to tensor of the given device, in bytes. Examples: .. code-block:: python >>> import paddle >>> paddle.device.max_memory_allocated('npu:0') >>> paddle.device.max_memory_allocated('npu') >>> paddle.device.max_memory_allocated(0) >>> paddle.device.max_memory_allocated(Paddle.CustomPlace('npu',0)) rN:rInvalid device ID '4'. After colon must be digits only. Example: 'npu:0' The input: z is not expected. Because paddle.device.max_memory_allocated only support str, int or CustomPlace. Please input appropriate device again! Example: 'npu:0' Allocated isinstancestrrfindisdigit ValueErrorintrr get_device_iddevice_memory_stat_peak_valuedevice device_id colon_idx device_id_strs r!rrns(I ~ FC LL% ?I"q=?3M ((** )-9'' M*I FC  FD,, - -((* &"     - -k9 EEr"cSnUcSnO[U[5(aQURS5nUS:XaSnOXS-SnUR5(d[ SUS35e[ U5nOW[U[ 5(aUnO?[U[ R5(aUR5nO[ SUS 35e[ R"S U5$) a Return the peak size of memory that is held by the allocator of the given device. Args: device(_CustomPlaceLike, optional): Support input like 'npu:0', 'mlu', int, or CustomPlace. If None, the device is the first available custom device with index 0. Returns: int: The peak size of memory that is held by the allocator of the given device, in bytes. Examples: .. code-block:: python >>> import paddle >>> paddle.device.max_memory_reserved('npu:0') >>> paddle.device.max_memory_reserved('npu') >>> paddle.device.max_memory_reserved(0) >>> paddle.device.max_memory_reserved(Paddle.CustomPlace('npu',0)) rNr'r(rr)r*r+z is not expected. Because paddle.device.max_memory_reserved only support str, int or CustomPlace. Please input appropriate device again! Example: 'npu:0'Reservedr-r6s r!rrs(I ~ FC LL% ?I"q=?3M ((** )-9'' M*I FC  FD,, - -((* &"     - -j) DDr"cSnUcSnO[U[5(aQURS5nUS:XaSnOXS-SnUR5(d[ SUS35e[ U5nOW[U[ 5(aUnO?[U[ R5(aUR5nO[ SUS 35e[ R"S U5 g) aK Reset the peak size of memory that is allocated to tensor of the given device. Args: device(_CustomPlaceLike, optional): Support input like 'npu:0', 'mlu', int, or CustomPlace. If None, the device is the first available custom device with index 0. Examples: .. code-block:: python >>> import paddle >>> paddle.device.reset_max_memory_allocated('npu:0') >>> paddle.device.reset_max_memory_allocated('npu') >>> paddle.device.reset_max_memory_allocated(0) >>> paddle.device.reset_max_memory_allocated(Paddle.CustomPlace('npu',0)) rNr'r(rr)r*r+z is not expected. Because paddle.device.reset_max_memory_allocated only support str, int or CustomPlace. Please input appropriate device again! Example: 'npu:0'r, r.r/r0r1r2r3rr r4#device_memory_stat_reset_peak_valuer6s r!rrs"I ~ FC LL% ?I"q=?3M ((** )-9'' M*I FC  FD,, - -((* &"    ,,[)Dr"cSnUcSnO[U[5(aQURS5nUS:XaSnOXS-SnUR5(d[ SUS35e[ U5nOW[U[ 5(aUnO?[U[ R5(aUR5nO[ SUS 35e[ R"S U5 g) aI Reset the peak size of memory that is held by the allocator of the given device. Args: device(_CustomPlaceLike, optional): Support input like 'npu:0', 'mlu', int, or CustomPlace. If None, the device is the first available custom device with index 0. Examples: .. code-block:: python >>> import paddle >>> paddle.device.reset_max_memory_reserved('npu:0') >>> paddle.device.reset_max_memory_reserved('npu') >>> paddle.device.reset_max_memory_reserved(0) >>> paddle.device.reset_max_memory_reserved(Paddle.CustomPlace('npu',0)) rNr'r(rr)r*r+z is not expected. Because paddle.device.reset_max_memory_reserved only support str, int or CustomPlace. Please input appropriate device again! Example: 'npu:0'r<r>r6s r!rrs"I ~ FC LL% ?I"q=?3M ((** )-9'' M*I FC  FD,, - -((* &"    ,,ZCr"cSnUcSnO[U[5(aQURS5nUS:XaSnOXS-SnUR5(d[ SUS35e[ U5nOW[U[ 5(aUnO?[U[ R5(aUR5nO[ SUS 35e[ R"S U5$) a Return the current size of memory that is allocated to tensor of the given device. Args: device(_CustomPlaceLike, optional): Support input like 'npu:0', 'mlu', int, or CustomPlace. If None, the device is the first available custom device with index 0. Returns: int: The current size of memory that is allocated to tensor of the given device, in bytes. Examples: .. code-block:: python >>> import paddle >>> paddle.device.memory_allocated('npu:0') >>> paddle.device.memory_allocated('npu') >>> paddle.device.memory_allocated(0) >>> paddle.device.memory_allocated(Paddle.CustomPlace('npu',0)) rNr'r(rr)r*r+z is not expected. Because paddle.device.memory_allocated only support str, int or CustomPlace. Please input appropriate device again! Example: 'npu:0'r, r.r/r0r1r2r3rr r4 device_memory_stat_current_valuer6s r!rr8s(I ~ FC LL% ?I"q=?3M ((** )-9'' M*I FC  FD,, - -((* &"     0 0i HHr"cSnUcSnO[U[5(aQURS5nUS:XaSnOXS-SnUR5(d[ SUS35e[ U5nOW[U[ 5(aUnO?[U[ R5(aUR5nO[ SUS 35e[ R"S U5$) a Return the current size of memory that is held by the allocator of the given device. Args: device(_CustomPlaceLike, optional): Support input like 'npu:0', 'mlu', int, or CustomPlace. If None, the device is the first available custom device with index 0. Returns: int: The current size of memory that is held by the allocator of the given device, in bytes. Examples: .. code-block:: python >>> import paddle >>> paddle.device.memory_reserved('npu:0') >>> paddle.device.memory_reserved('npu') >>> paddle.device.memory_reserved(0) >>> paddle.device.memory_reserved(Paddle.CustomPlace('npu',0)) rNr'r(rr)r*r+z is not expected. Because paddle.device.memory_reserved only support str, int or CustomPlace. Please input appropriate device again! Example: 'npu:0'r<rBr6s r!rrls(I ~ FC LL% ?I"q=?3M ((** )-9'' M*I FC  FD,, - -((* &"     0 0Y GGr"cSnUcSnO[U[5(aQURS5nUS:XaSnOXS-SnUR5(d[ SUS35e[ U5nOW[U[ 5(aUnO?[U[ R5(aUR5nO[ SUS 35e[ R"[U5$) a% Return the current stream by the device. Args: device(_CustomPlaceLike, optional): Support input like 'npu:0', 'mlu', int, or CustomPlace. If None, the device is the first available custom device with index 0. Returns: Stream: The stream to the device. Examples: .. code-block:: python >>> import paddle >>> paddle.device.current_stream('npu:0') >>> paddle.device.current_stream('npu') >>> paddle.device.current_stream(0) >>> paddle.device.current_stream(Paddle.CustomPlace('npu',0)) rNr'r(rr)r*r+z is not expected. Because paddle.device.current_stream only support str, int or CustomPlace. Please input appropriate device again! Example: 'npu:0') r.r/r0r1r2r3rr r4!_get_current_custom_device_streamrr6s r!rrs(I ~ FC LL% ?I"q=?3M ((** )-9'' M*I FC  FD,, - -((* &"     1 1(I FFr"cSnUcSnO[U[5(aQURS5nUS:XaSnOXS-SnUR5(d[ SUS35e[ U5nOW[U[ 5(aUnO?[U[ R5(aUR5nO[ SUS 35e[ R"[U5 g) a Wait for the compute on the given device to finish. Args: device(_CustomPlaceLike, optional): Support input like 'npu:0', 'mlu', int, or CustomPlace. If None, the device is the first available custom device with index 0. Examples: .. code-block:: python >>> import paddle >>> paddle.device.synchronize('npu:0') >>> paddle.device.synchronize('npu') >>> paddle.device.synchronize(0) >>> paddle.device.synchronize(Paddle.CustomPlace('npu',0)) rNr'r(rr)r*r+z is not expected. Because paddle.device.synchronize only support str, int or CustomPlace. Please input appropriate device again! Example: 'npu:0') r.r/r0r1r2r3rr r4_synchronize_custom_devicerr6s r!rrs"I ~ FC LL% ?I"q=?3M ((** )-9'' M*I FC  FD,, - -((* &"    ##Hi8r"c [RRU5n[U[R 5(a#[R "5R5$[R"U5R5$)a Get the random state for the default generator. Returns: Tensor: The random state tensor. Examples: .. code-block:: python >>> # doctest: +REQUIRES(env:CUSTOM_DEVICE) >>> import paddle >>> paddle.device.get_rng_state() ) paddler7device_to_placer.rCPUPlacedefault_cpu_generator get_statedefault_custom_device_generator)r7places r! get_rng_staterQs[$ MM ) )& 1E%''))+5577  / / 6 @ @ BBr"c[RRU5n[U[R 5(a%[R "5RU5 g[R"U5RU5 g)a& Set the random number generator state of the specified device. Args: new_state (core.GeneratorState): The desired RNG state to set. This should be a state object previously obtained from ``get_rng_state()``. device (DeviceLike, optional): The device to set the RNG state for. If not specified, uses the current default device (as returned by ``paddle.framework._current_expected_place_()``). Can be a device object, integer device ID, or device string. Returns: None Examples: .. code-block:: python >>> import paddle >>> # Save RNG state >>> state = paddle.device.get_rng_state() >>> # Do some random operations >>> x = paddle.randn([2, 3]) >>> # Restore RNG state >>> paddle.device.set_rng_state(state) N) rJr7rKr.rrLrM set_staterO) new_stater7rPs r! set_rng_staterUsY6 MM ) )& 1E%'' ""$..y9 ,,U3==iHr"c([U5n[RR5n[ U[ R 5(a%[ R"5RU5 g[ R"U5RU5 g)aSet the seed for generating random numbers for the current Device. .. warning:: If you are working with a multi-Device model, this function is insufficient to get determinism. To seed all Devices, use :func:`manual_seed_all`. If current Device is CPU, this function will set the seed of the default CPU generator. Sets the seed for global default generator, which manages the random number generation. Args: seed(int): The random seed to set. Returns: None Examples: .. code-block:: python >>> # doctest: +REQUIRES(env:CUSTOM_DEVICE) >>> import paddle >>> paddle.device.manual_seed(102) >>> # paddle.cuda.manual_seed(102) is equivalent to paddle.device.manual_seed(102) N) r3rJ framework_current_expected_placer.rrLrM manual_seedrO)seedrPs r!rYrY?sb0 t9D    4 4 6E%'' ""$006 ,,U3??Er")N)rz str | Nonereturnr3)r[None)r7_CustomPlaceLike | Noner[r3)r7r]r[r\)r7r]r[zcore.CustomStream)r7r]r[core.GeneratorState)rTr^r7r]r[r\)rZr3r[r\)* __future__rtypingrrtyping_extensionsrrJ paddle.basercustom_streamsr r r r r r/r3r__annotations__get_all_custom_device_type dev_typesris_compiled_with_custom_device Exception gpgpu_backendrdefault_backend__all__rrrrrrrrrrrQrUrYr%r"r!rls/#'' ""'   #i  + + - $9Q<$ D77AA V  9946 "6 1Fh1Eh.Eb.Db1Ih1Hh1Gh.9d'+C #CC2GKI"I,CI IDFr"