Pi hddlZddlZddlmZdedejdedededef dZdeeeffd Z dS) Nseedgenerator_rankrank world_sizeepochreturnc|||z|zttjdd|dzf|dS)Nrl)size generator) manual_seedinttorchrandintitem)rrrrrs r/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchdata/nodes/samplers/utils.py_get_rank_seedr sYtj047888 u}Q 519,.YYYZ\]bbdd e eectjr:tjr'tjtj}}ntt jdd}t jdd} t|}t|}n#t$rd}d}YnwxYw||ks|dkrtd|d|dz d ||fS) av Returns the rank and world size of the current process. If distributed is initialized, returns the rank and world size from the distributed environment. If distributed is not initialized, returns the rank and world size from the environment variables. If neither distributed nor environment variables are set, returns a rank of 0 and a world size of 1. RANK0 WORLD_SIZE1rr z Invalid rank z%, rank should be in the interval [0, ]) dist is_availableis_initializedget_rankget_world_sizeosenvirongetr ValueError)rr_rank _world_sizes rget_rank_and_world_sizer's  t244 =??D,?,A,Aj vs++jnn\377  u::D[))JJ   DJJJ  zTAXXeeeT^abTbeeefff  sB..B?>B?) r!rtorch.distributed distributedrr Generatorrtupler'rrr-s ffeofSfVYfbefjmffff sCxr