o Ne$ć@sPddlmZddlmZgZdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dS)éN)ŚcorecCs@t|ƒ}t ”rtt ”ƒD] }t |” |”qt ” |”S)av 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) )ŚintrŚis_compiled_with_cudaŚrangeŚget_cuda_device_countŚdefault_cuda_generatorZ manual_seedZdefault_cpu_generator)ŚseedŚi©r śGD:\Projects\ConvertPro\env\Lib\site-packages\paddle/framework/random.pyrs rcCs6g}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() )rrrrŚappendrZ get_state©Z state_listr r r r Śget_cuda_rng_state4s rcCsNt ”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)rrŚlenrŚ ValueErrorrrZ set_stater r r r Śset_cuda_rng_stateNs’śrcCs*|t ”_|t ”_t ”}| |”dS)aW 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)ŚfluidZdefault_main_programZ random_seedZdefault_startup_programZProgramZ global_seed)rŚprogramr r r Ś_manual_program_seedjs rcCst ||”dS©N)rŚset_random_seed_generator)Śnamerr r r r}srcCs t |”Sr)rŚget_random_seed_generator)rr r r rs r) Z paddle.fluidrrŚ__all__rrrrrrr r r r Śs