9iddlZddlmZmZmZddlZddlZddlm Z ddl m Z ddl m Z mZddlmZmZmZed r ddlmcmZdd ed efd Zddeedeej4fdZddeeeefdeej4fdZy)N)ListOptionalUnion)AcceleratorState)CUDA_DISTRIBUTED_TYPES)DistributedTypeRNGType)is_npu_availableis_tpu_availableis_xpu_availableF) check_deviceseeddevice_specificc|r|tjz }tj|tjj|t j |tr t jj|nItr t jj|nt jj|trtj|yy)a= Helper function for reproducible behavior to set the seed in `random`, `numpy`, `torch`. Args: seed (`int`): The seed to set. device_specific (`bool`, *optional*, defaults to `False`): Whether to differ the seed on each device slightly with `self.process_index`. N)r process_indexrandomrnptorch manual_seedrxpumanual_seed_allr npucudar xm set_rng_state)rrs a/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/accelerate/utils/random.pyset_seedrs  "000 KKIINN4 d !!$'   !!$' ""4( rng_type generatorc|tjk(rtj}n0|tjk(rtj j}n|tj k(r9tsJdtjtj}n|tjk(r0tsJdtjj}no|tjk(r0tsJdtjj}n,|tj k(r|Jd|j#}t%}|j&t(j*k(r^j-tj.}tj0|gtj2|j5}n|j&t6vs:|j&t(j8k(s|j&t(j:k(rLj-|j<}tj>jA|d|j5}n=|j&t(jBk(r tj>jAd|tjk(rtjDy|tjk(r tj jEy|tjk(r tjjEy|tjk(r tjjEy|tj k(r$tjDjGy|tj k(r|jIyy)Nz;Can't synchronize XLA seeds on an environment without TPUs.z;Can't synchronize NPU seeds on an environment without NPUs.z;Can't synchronize XPU seeds on an environment without XPUs.z)Need a generator to synchronize its seed.r)%r TORCHr get_rng_stateCUDArXLAr tensorrNPUr rXPUrr GENERATOR get_staterdistributed_typer TPUto xla_devicecollective_broadcast mark_stepcpur MULTI_NPU MULTI_XPUdevice distributed broadcast MULTI_CPUritem set_state)r!r" rng_statestates rsynchronize_rng_stater>9s7== '') W\\ !JJ,,. W[[ !`#``!LL!1!1!34 W[[ !`#``!II++- W[[ !`#``!II++- W&& &$Q&QQ$'')   E !4!44LL1   , MMO  "88  ! !_%>%> >  ! !_%>%> >LL.  ##Iq1MMO   ?#<#< < ##Iq17==  I& W\\ !   + W[[   * W[[   * W[[  )* W&& &I& 'r rng_typesc>|D]}tt||y)N)r")r>r )r?r"r!s rsynchronize_rng_statesrAms"Fgh/9EFr )F)NN)N)rtypingrrrnumpyrrr=r constantsr dataclassesr r importsr r rtorch_xla.core.xla_modelcore xla_modelrintboolr Generatorr>strrAr rrOs(( $-1II'))341'HW$51'RWRaRaIb1'hFd5g+>&?FHUZUdUdLeFr