x-j)^ddlmZddlmZmZddlZddlmZddlmZer"ddl m Z ddlm Z m Z ddl mZGd d e ZgZe d&d'dZe d&d(dZe d&d)dZd*dZe d&d+dZe d&d,d Ze d&d-d"Z d*d#Z d.d/d%ZdS)0) annotations) TYPE_CHECKINGoverloadN)base)core)Sequence)LiteralProtocol) PlaceLikeceZdZddZdS)_GeneratorStatereturnintcdSN)selfs `/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/incubate/framework/random.py current_seedz_GeneratorState.current_seedsN)rr)__name__ __module__ __qualname__rrrrr r s******rr .devicePlaceLike | None use_index Literal[True]r list[int]cdSrrrrs r get_rng_stater!$s rLiteral[False]list[_GeneratorState]cdSrrr s rr!r!*s  Crbool!list[int] | list[_GeneratorState]cdSrrr s rr!r!0s ),rFc hfd}g}|tjntj|t tjr1| |tj nt tj rUttj D]2}| |tj|3n.t tjrTttjD]2}| |tj|3nt tjrt'fdtjD}t|D]W}| |tjtj|Xnt/d|S)aH Get all random states of random generators of specified device. Args: device(str): This parameter determines the specific running device. It can be ``cpu``, ``gpu``, ``xpu``, Default is None. If None, return the generators of current device (specified by ``set_device``). use_index(bool): If use index is True, return the index that saved in the generator Returns: GeneratorState: object. Examples: .. code-block:: python >>> import paddle >>> sts = paddle.incubate.get_rng_state() cXr|S|Sr)get_state_index get_state) generatorrs rr+z get_rng_state..get_stateFs/  ),,.. .&&(( (rNcrg|]3}|ddk4S):r)get_device_typesplit).0splaces r z!get_rng_state..\sH   %%''1773<<?:   rz5get_rng_state is not implemented for current device: )r framework_current_expected_place_paddler_convert_to_place isinstancerCPUPlaceappenddefault_cpu_generator CUDAPlacerangeget_cuda_device_countdefault_cuda_generatorXPUPlaceget_xpu_device_countdefault_xpu_generator CustomPlacesumget_available_custom_devicedefault_custom_device_generatorr/ ValueError)rrr+ state_listidev_cntr3s ` @rr!r!6sO ))))) J ~7799 //77%'' ))D$>$@$@AABBBB E4> * * t13344 I IA   ii(CA(F(FGG H H H H I E4= ) ) t02233 H HA   ii(B1(E(EFF G G G G H E4+ , ,     9;;      w  A    8()>)>)@)@!DD      KE K K    rrI Sequence[int]NonecdSrrrIrrs r set_rng_staterPq 3rSequence[_GeneratorState]cdSrrrOs rrPrPyrQr)Sequence[int] | Sequence[_GeneratorState]cdSrrrOs rrPrPrQrc Pfd}|tj}n||}t |t jr|t|t jkstdtt jD]&}|t j |||'dSt |t j r|t|t j kstdtt j D]&}|t j|||'dSt |t jrt j}|d}t j|}t||kstd|dt|D]K}|t jt j||||LdSt |t jrGt|dkstd |t j|ddStd |) a Sets generator state for all device generators. Args: state_list(list|tuple): The device states to set back to device generators. state_list is obtained from get_rng_state(). device(str): This parameter determines the specific running device. It can be ``cpu``, ``gpu``, ``xpu``, Default is None. If None, return the generators of current device (specified by ``set_device``). use_index(bool): If use index is True, state_list should be the indices of the states Returns: None. Examples: .. code-block:: python >>> import paddle >>> sts = paddle.incubate.get_rng_state() >>> paddle.incubate.set_rng_state(sts) cdr||dS||dSr)set_state_index set_state)r,staters rrYz set_rng_state..set_states?  '  % %e , , , , ,    & & & & &rN@Length of gpu state list should be equal to the gpu device count@Length of xpu state list should be equal to the xpu device countr:Length of custom device state list should be equal to the device count-Length of cpu state list should be equal to 1z5set_rng_state is not implemented for current device: )rr5r6r8r9rr=lenr?rHr>r@rArBrCrDget_all_custom_device_typeget_custom_device_countrGr/r:r<) rIrrrYr3rJ dev_typesdev_typerKs ` rrPrPs8''''' ~7799((00%(($ :$"<">">>>R t13344 E EA Id1!44jm D D D D E E E4= ) ) :$";"="===R t02233 D DA Id033Z] C C C C D D E4+ , , 355 Q<.x88:'))cWccc w  A I4$U%:%:%<%>>>> KE K K   r Sequence[_GeneratorState] | Nonec Jg}|tj}n||}|t |}t |t jrt|t j kstdtt j D]B}| t j |||Cn't |t jrt|t jkstdtt jD]B}| t j|||Cnut |t jrt j}|d}t j|}t||kstd|dt|D]g}| t jt j||||hnt |t jrbt|dkstd| t j|dntd ||S) ax The register_rng_state_as_index function creates and registers a new generator state within the generator. It enables users to manage multiple generator states via indices, offering a convenient way to switch between these states without directly manipulating the generator's state. Args: state_list(list|tuple): A list or tuple representing the RNG states for devices. If not provided, the function will register the current state. device(str): This parameter determines the specific running device. It can be ``cpu``, ``gpu``, ``xpu``, Default is None. If None, return the generators of current device (specified by ``set_device``). Returns: A list of indices representing the positions at which the new states were saved within the generator. These indices can be used to switch between states using set_rng_state(use_index=True) Examples: .. code-block:: python >>> import paddle >>> old_index = paddle.incubate.get_rng_state(use_index=True) >>> print(old_index) [0] >>> new_index = paddle.incubate.register_rng_state_as_index() >>> print(new_index) [1] >>> paddle.incubate.set_rng_state(old_index, use_index=True) >>> paddle.incubate.set_rng_state(new_index, use_index=True) Nr[r\rr]r^r_r`z@register_rng_state_index is not implemented for current device: )rr5_current_expected_placer8r!r9rr=rar?rHr>r;r@register_state_indexrArBrCrDrbrcrGr/r:r<)rIrnew_state_index_listr3rJrdrerKs rregister_rng_state_as_indexrks#H ~6688((00"6** %((- :$"<">">>>R t13344  A ' '+A..CCqM      E4= ) )" :$";"="===R t02233  A ' '*1--BBqM      E4+ , , 355 Q<.x88:'))cWccc w  A ' '4$U%:%:%<%rrs:#"""""******** +((((((((((((((((((((+++++(+++  ?B   @C       69,,,, , 8888v  #"   # #   #  K K K K ^48#\ \ \ \ \ \ \ r