\;j!\ddlZddlmZddlmZgZdZd dZdZd dZdZ d Z d Z d Z y) N)base)corec t|}tjrGttjD]&}tj |j |(nZtjrFttjD]&}tj|j |(tjj}t|tjrttj Dcgc]'}|j#|j%ddk()c}}t|D]H}tj&tj|j#|j |Jtj(j |Scc}w)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) :r)intris_compiled_with_cudarangeget_cuda_device_countdefault_cuda_generator manual_seedis_compiled_with_xpuget_xpu_device_countdefault_xpu_generatorr framework_current_expected_place isinstance CustomPlacesumget_available_custom_deviceget_device_typesplitdefault_custom_device_generatordefault_cpu_generator)seediplacesdev_cnts `G:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/framework/random.pyrrsU* t9D !!#t1134A  ' ' * 6 6t <5 " " $t0023A  & &q ) 5 5d ;4 NN 2 2 4E%))*99; ;A%%'1773<?:;   wA  0 0  !6!6!8!< k$   % % ' 3 3D 99 s,Gc g}|tjj}ntjj |}t |tjr3|jtjj|St |tjrVttjD]4}|jtj|j6|St |tj rVttj"D]4}|jtj$|j6|St |tj&rt)tj*Dcgc]'}|j-|j/ddk()c}}t|D]V}|jtj0tj&|j-|jX|St3d|cc}w)a 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``). Returns: GeneratorState: object. Examples: .. code-block:: python >>> import paddle >>> sts = paddle.get_rng_state() rrz5get_rng_state is not implemented for current device: )rrrpaddledevice_convert_to_placerrCPUPlaceappendr get_state CUDAPlacer r r XPUPlacerrrrrrrr ValueError)r" state_listrrrrs r get_rng_stater+Cs$J ~668 //7%'$446@@BC2 1 E4>> *t1134A   d99!<FFH I5. + E4== )t0023A   d88;EEG H4( % E4++ ,99; ;A%%'1773<?:;   wA   44$$U%:%:%>> import paddle >>> sts = paddle.get_cuda_rng_state() )rrr r r%r r&r*rs rget_cuda_rng_stater.xsW$J !!#t1134A   d99!<FFH I5 c V|tjj}n|j|}t |t j rut|t jk(s tdtt jD])}t j|j||+yt |t jrut|t jk(s tdtt jD])}t j|j||+yt |t j rt#t j$Dcgc]'}|j'|j)ddk()c}}t||k(std|j+dt|D]K}t j,t j |j'|j||Myt |t j.r@t|dk(s td t j0j|dytd |cc}w) 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 shoule be equal to the gpu device countz@Length of xpu state list shoule be equal to the xpu device countrrz:Length of custom device state list shoule be equal to the z device countz-Length of cpu state list shoule be equal to 1z5set_rng_state is not implemented for current device: )rrrr#rrr'lenr r)r r set_stater(rrrrrrrget_dtype_typerr$r)r*r"rrrrs r set_rng_stater5s/,~668((0%(:$"<"<">>R t1134A  ' ' * 4 4Z] C5 E4== ):$";";"==R t0023A  & &q ) 3 3JqM B4 E4++ ,99; ;A%%'1773<?:;   :')LUMaMaMcLddqr wA  0 0  !6!6!8!< i 1 &  E4== ):!#LM M ""$..z!}=CE7 K  % s,J&ctjrut|tjk(s t dt tjD])}tj |j||+yy)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 shoule be equal to the cuda device countN)rrr2r r)r r r3r-s rset_cuda_rng_stater7sq& !!#:$"<"<">>T t1134A  ' ' * 4 4Z] C5 $r/c|tjj_|tjj _tjj }|j |y)aQ Sets global seed for generating random numbers. NOTE(zhiqiu): This is the original implemention 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)r!staticdefault_main_program random_seeddefault_startup_programProgram global_seed)rprograms r_manual_program_seedr@sN8FMM))+7mm##%G r/c0tj||yN)rset_random_seed_generator)namers rrCrCs""4.r/c,tj|SrB)rget_random_seed_generator)rDs rrFrFs  ) )$ //r/rB) r!r paddle.baser__all__rr+r.r5r7r@rCrFr/rrJsA  ):X2j4? DD8&/0r/