zj&ddlmZddlmZddlZddlmZerddlmZgZ dd Z dddZ ddZ dd dZ d!dZd"dZd#dZd$dZGddZdS)%) annotations) TYPE_CHECKINGN)core)Sequenceseedintreturnpaddle.base.core.Generatorcht|}tjrKtt jD])}t j||*n]tjrJtt j D])}t j ||*tj ttjrtfdt jD}t|D]N}t jtj||Ot j|S)a~ Sets the seed for global default generator, which manages the random number generation. Args: seed(int): The random seed to set. It is recommend to set a large int number. Returns: Generator: The global default generator object. Examples: .. code-block:: python >>> import paddle >>> gen = paddle.seed(102) crg|]3}|ddk4S:rget_device_typesplit.0splaces g/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/framework/random.py zseed..<H   %%''1773<<?:   )rpaddleis_compiled_with_cudarangerget_cuda_device_countdefault_cuda_generator manual_seedis_compiled_with_xpuget_xpu_device_countdefault_xpu_generator framework_current_expected_place isinstance CustomPlacesumget_available_custom_devicedefault_custom_device_generatorrdefault_cpu_generator)ridev_cntrs @rrrs* t99D #%%<t13344 = =A  ' * * 6 6t < < < < =  $ & &<t02233 < >  k$  % ' ' 3 3D 9 99rdevice str | None%list[paddle.base.core.GeneratorState]c g}|tjntj|t tjr:|tj nt tj r^ttj D];}|tj| <n@t tjr]ttjD];}|tj| >> import paddle >>> sts = paddle.get_rng_state() Ncrg|]3}|ddk4Sr rrs rrz!get_rng_state..lrrz5get_rng_state is not implemented for current device: )rr#_current_expected_place_r-_convert_to_placer%CPUPlaceappendrr* get_state CUDAPlacerrrXPUPlacer!r"r&r'r(r)r ValueError)r- state_listr+r,rs @r get_rng_stater;Hs7(J ~ 99;; //77%)) $466@@BBCCCC E6+ , , t13344 J JA   d9!<<FFHH I I I I J E6? + + t02233 I IA   d8;;EEGG H H H H I E6- . .     9;;      w  A   4$U%:%:%<%>> import paddle >>> sts = paddle.get_cuda_rng_state() )rrrrrr5rr6r:r+s rget_cuda_rng_stater>sq$J #%%Jt13344 J JA   d9!<<FFHH I I I I rr:)Sequence[paddle.base.core.GeneratorState]Nonec|tj}ntj|}t |tjrt|tj kstdttj D]/}tj | ||0dSt |tjrt|tjkstdttjD]/}tj| ||0dSt |tjrtj}|d}tj|}t||kstd|dt|D]T}tjtj|| ||UdSt |tjrPt|dkstdtj |ddStd |) aq 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``). Returns: None. Examples: .. code-block:: python >>> import paddle >>> sts = paddle.get_rng_state() >>> paddle.set_rng_state(sts) Nz@Length of gpu state list should be equal to the gpu device countz@Length of xpu state list should be equal to the xpu device countrz:Length of custom device state list should be equal to the z device countz-Length of cpu state list should be equal to 1z5set_rng_state is not implemented for current device: )rr#r2r-r3r%r7lenrrr9rr set_stater8r!r"r&get_all_custom_device_typeget_custom_device_countr)rr4r*)r:r-rr+ dev_typesdev_typer,s r set_rng_staterIs2~ 99;; //77%)**! :$"<">">>>R t13344 D DA  ' * * 4 4Z] C C C C D D E6? + + :$";"="===R t02233 C CA  &q ) ) 3 3JqM B B B B C C E6- . . 355 Q<.x88:'))cWccc w ' 'A  0"5#8#8#:#:A>>  i 1 &&&& ' ' E4= ) ) :!##LMM M "$$..z!}===== KE K K   rc6tjrt|tjkst dt tjD]1}tj|||0dSdS)a Sets generator state for all cuda generators. Args: state_list(list|tuple): The cuda states to set back to cuda generators. state_list is obtained from get_cuda_rng_state(). Returns: None. Examples: .. code-block:: python >>> import paddle >>> sts = paddle.get_cuda_rng_state() >>> paddle.set_cuda_rng_state(sts) zBLength of cuda state list should be equal to the cuda device countN) rrrCrrr9rrrDr=s rset_cuda_rng_staterKs*#%%D:$"<">">>>T t13344 D DA  ' * * 4 4Z] C C C C DD  D Drc|tj_|tj_tj}||dS)aS Sets global seed for generating random numbers. NOTE(zhiqiu): This is the original implementation of seed. Keeps it temporally since CUDA generator is not developed, so we need it in the unittest. Args: seed(int): The random seed to set. It is recommend to set a large int number. Returns: None N)rstaticdefault_main_program random_seeddefault_startup_programProgram global_seed)rprograms r_manual_program_seedrTs]8FM))++7m##%%G rnamestrc0tj||dSN)rset_random_seed_generator)rUrs rrYrY s"4.....rc*tj|SrX)rget_random_seed_generator)rUs rr[r[s  )$ / //rceZdZ dddZdS) GeneratorNr-str | int | paddle.core.Placer core.Generatorctj|}t|tjrt jSt|tjr&t j| St|tj r&t j | St|tj rt j |SdS)a Generator is a random number generator. Args: device(str|int|paddle.core.Place): The device type to create the generator on. It can be ``cpu``, ``gpu``, ``xpu``, or a paddle.core.Place instance. default is None, which means using current device. Examples: .. code-block:: python >>> import paddle >>> g_cpu = paddle.Generator() N)rr-device_to_placer%rr4r*r7r gpu_device_idr8r"r&r))clsr-rs r__new__zGenerator.__new__s" --f55 eT] + + ?-// / t~ . . ?.u/B/B/D/DEE E t} - - ?-e.A.A.C.CDD D t/ 0 0 ?7>> > ? ?rrX)r-r^r r_)__name__ __module__ __qualname__rdrrr]r]s059???????rr])rrr r rX)r-r.r r/)r r/)r:r?r-r.r r@)r:r?r r@)rrr r@)rUrVrrr r@)rUrVr r ) __future__rtypingrr paddle.basercollections.abcr__all__rr;r>rIrKrTrYr[r]rhrrrnsR#""""" )(((((( ):):):):Z44444n8? ? ? ? ? DDDDD<&////0000??????????r