o "Õjŗ ć@slddlZddlmZddlmZgZdd„Zddd„Zdd „Zdd d „Zd d „Z dd„Z dd„Z dd„Z dS)éN)Śbase)ŚcorecsĘt|ƒ}t ”rtt ”ƒD] }t |” |”qnt ”r/tt ”ƒD] }t  |” |”q$t j   ”‰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©śX/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/framework/random.pyŚ 7ó’’zseed..)ŚintrŚis_compiled_with_cudaŚrangeŚget_cuda_device_countŚdefault_cuda_generatorZ manual_seedZis_compiled_with_xpuŚget_xpu_device_countŚdefault_xpu_generatorrŚ frameworkŚ_current_expected_placeŚ isinstanceŚ CustomPlaceŚsumŚget_available_custom_deviceŚdefault_custom_device_generatorrŚdefault_cpu_generator)ŚseedŚiŚdev_cntrr rr!s*’   ž’ ’ r!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: GeneratorState: object. Examples: .. code-block:: python >>> import paddle >>> sts = paddle.get_rng_state() Ncrrrr r rrrerz!get_rng_state..z5get_rng_state is not implemented for current device: )rrrŚpaddleŚdeviceŚ_convert_to_placerrŚCPUPlaceŚappendr Ś get_stateŚ CUDAPlacerrrŚXPUPlacerrrrrrrŚ ValueError)r%Ś state_listr"r#rr rŚ get_rng_stateCs@    č ė ī ž’ ’ż ü’r.cCs6g}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() )rrrrr(rr)©r-r"rrrŚget_cuda_rng_statexs r0csv|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 shoule be equal to the gpu device countz@Length of xpu state list shoule be equal to the xpu device countcrrrr r rrr½rz!set_rng_state..z:Length of custom device state list shoule be equal to the z device countéz-Length of cpu state list shoule be equal to 1rz5set_rng_state is not implemented for current device: )rrrr&rrr*Ślenrr,rrŚ set_stater+rrrrrZget_dtype_typerrr'r )r-r%r"r#rr rŚ set_rng_state’sT   ’’ ’’  ž’ ’ ’ż  ’r4cCsNt ”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 shoule be equal to the cuda device countN)rrr2rr,rrr3r/rrrŚset_cuda_rng_stateŌs’śr5cCs0|tj ”_|tj ”_tj ”}| |”dS)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$ZstaticZdefault_main_programZ random_seedZdefault_startup_programZProgramZ global_seed)r!ŚprogramrrrŚ_manual_program_seedšs  r7cCst ||”dS©N)rŚset_random_seed_generator)Śnamer!rrrr9sr9cCs t |”Sr8)rŚget_random_seed_generator)r:rrrr;s r;r8) r$rZ paddle.baserŚ__all__r!r.r0r4r5r7r9r;rrrrŚs   ,5 B