o ńp i‡"ć@sœddlmZddlmZddlZddlmZerddlmZgZ d!d d„Z d"d#dd„Z d$dd„Z d"d%dd„Z d&dd„Zd'dd„Zd(dd„Zd)dd „ZdS)*é)Ś annotations)Ś TYPE_CHECKINGN)Ścore)ŚSequenceŚseedŚintŚreturnśpaddle.base.core.GeneratorcsĘt|ƒ}t ”rtt ”ƒD] }t |” |”qnt ”r/tt  ”ƒD] }t  |” |”q$tj   ”‰t ˆtjƒr\t‡fdd„t ”Dƒƒ}t|ƒD]}t t ˆ ”|”” |”qKt ” |”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) có"g|] }ˆ ”| d”dk‘qS©ś:r©Śget_device_typeŚsplit©Ś.0Śs©Zplace©ś^/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/framework/random.pyŚ <ó’’zseed..)rŚpaddleŚis_compiled_with_cudaŚrangerŚget_cuda_device_countŚdefault_cuda_generatorZ manual_seedZis_compiled_with_xpuŚget_xpu_device_countŚdefault_xpu_generatorŚ frameworkZ_current_expected_placeŚ isinstanceŚ CustomPlaceŚsumŚget_available_custom_deviceŚdefault_custom_device_generatorrŚdefault_cpu_generator)rŚiŚdev_cntrrrrs*’   ž’ ’ Śdeviceś str | Noneś%list[paddle.base.core.GeneratorState]c sg}|dur tj ”‰ntj |”‰tˆtjƒr#| t  ”  ””|Stˆtj ƒr>t t  ”ƒD] }| t |”  ””q/|StˆtjƒrYt t ”ƒD] }| t |”  ””qJ|Stˆtjƒr…t‡fdd„t ”Dƒƒ}t |ƒD]}| t t ˆ ”|””  ””qp|Stdˆ›ƒ‚)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: list[GeneratorState], object. Examples: .. code-block:: python >>> import paddle >>> sts = paddle.get_rng_state() Ncr r r rrrrrlrz!get_rng_state..z5get_rng_state is not implemented for current device: )rrŚ_current_expected_place_r(Ś_convert_to_placer ŚCPUPlaceŚappendrr%Ś get_stateŚ CUDAPlacerrrŚXPUPlacerrr!r"r#r$rŚ ValueError)r(Ś state_listr&r'rrrŚ get_rng_stateHs@    č ė ī ž’ ’ż ü’r4cCs6g}t ”rtt ”ƒD] }| t |” ””q |S)z÷ Get random state of cuda generators. Args: None. Returns: GeneratorState: object. Examples: .. code-block:: python >>> import paddle >>> sts = paddle.get_cuda_rng_state() )rrrrrr.rr/©r3r&rrrŚget_cuda_rng_states r6r3ś)Sequence[paddle.base.core.GeneratorState]ŚNonecsv|dur tj ”‰n| |”‰tˆtjƒr6t|ƒt ”ks!t dƒ‚t t ”ƒD] }t  |”  ||”q'dStˆtj ƒr]t|ƒt ”ksHt dƒ‚t t ”ƒD] }t |”  ||”qNdStˆtjƒr™t‡fdd„t ”Dƒƒ}t|ƒ|ks€t dˆ ”›dƒ‚t |ƒD]}t t ˆ ”|””  ||”q„dStˆtjƒr“t|ƒdks©t d ƒ‚t ”  |d ”dSt d ˆ›ƒ‚) 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 countcr r r rrrrrĒrz!set_rng_state..z:Length of custom device state list should be equal to the z device countéz-Length of cpu state list should be equal to 1rz5set_rng_state is not implemented for current device: )rrr+r,r r0Ślenrrr2rrŚ set_stater1rrr!r"r#Zget_dtype_typer$rr-r%)r3r(r&r'rrrŚ set_rng_state™sT   ’’ ’’  ž’ ’ ’ż  ’r<cCsNt ”r#t|ƒt ”kstdƒ‚tt ”ƒD]}t |” ||”qdSdS)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) rrr:rrr2rrr;r5rrrŚset_cuda_rng_stateŽs’śr=cCs0|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)rZstaticZdefault_main_programZ random_seedZdefault_startup_programZProgramZ global_seed)rŚprogramrrrŚ_manual_program_seedüs  r?ŚnameŚstrcCst ||”dS©N)rŚset_random_seed_generator)r@rrrrrCsrCcCs t |”SrB)rŚget_random_seed_generator)r@rrrrDs rD)rrrr rB)r(r)rr*)rr*)r3r7r(r)rr8)r3r7rr8)rrrr8)r@rArrrr8)r@rArr )Ś __future__rŚtypingrrZ paddle.baserŚcollections.abcrŚ__all__rr4r6r<r=r?rCrDrrrrŚs"     - ’ 7 ž E