o ॵiã@sZddlZz ddlmmZWn eydZYnwdd„Zdd„ZGdd„deƒZ dS)éNcCs<dt ¡i}tdurt ¡|d<tj ¡rtj ¡|d<|S)zC Get random number generator state of torch, xla and cuda. Útorch_rng_stateNÚ xla_rng_stateÚcuda_rng_state)ÚtorchÚ get_rng_stateÚxmÚcudaÚ is_available©Ústate©r úb/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/preprocessors/ofa/utils/random_help.pyr s   rcCsFt |d¡tdurt |d¡tj ¡r!tj |d¡dSdS)zC Set random number generator state of torch, xla and cuda. rNrr)rÚ set_rng_staterrr r r r r rs  ÿrc@s(eZdZdZdd„Zdd„Zdd„ZdS) Úset_torch_seedz1 Set random seed to torch, xla and cuda. cCsPt|tƒsJ‚tƒ|_t |¡tdurt |¡tj  ¡r&tj |¡dSdS©N) Ú isinstanceÚintrÚ rng_staterÚ manual_seedrrrr )ÚselfÚseedr r r Ú__init__'s   ÿzset_torch_seed.__init__cCs|Srr )rr r r Ú __enter__1szset_torch_seed.__enter__cGst|jƒdSr)rr)rÚexcr r r Ú__exit__4szset_torch_seed.__exit__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr r r r r"s  r) rÚtorch_xla.core.xla_modelÚcoreÚ xla_modelrÚ ImportErrorrrÚobjectrr r r r Ús ÿ