
    3j	                     >    S SK r S SKrS r\ R                  S 5       rg)    Nc                 :   [        5          [        R                  " S5        U " U0 UD6n[        U[        R                  5      (       a8  UR
                  R                  S:X  a  [        R                  R                  5         UsSSS5        $ ! , (       d  f       g= f)zWrapper to set seed manually for some functions like dropout
See: https://github.com/pytorch/pytorch/pull/62315#issuecomment-896143189 for more details.
*   lazyN)	freeze_rng_statetorchmanual_seed
isinstanceTensordevicetype_lazy	mark_step)opargskwargsoutputs       N/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/testing/_utils.pywrapper_set_seedr      sj     
	"T$V$fell++0B0Bf0L KK!!# 
		s   A7B
Bc            	   #   ,  #    [         R                  R                  R                  5          [         R                  R                  5          [         R                  " 5       n [         R                  R                  5       (       aN  [         R                  R                  SS9nUb.  [         R                  " UR                  5      R                  5       nS S S 5        S S S 5         S v   [         R                  R                  R                  5          [         R                  R                  5          [         R                  R                  5       (       aO  [         R                  R                  SS9nUb/  [         R                  " UR                  5      R                  W5        [         R                  " W 5        S S S 5        S S S 5        g ! , (       d  f       N= f! , (       d  f       GN= f! , (       d  f       N:= f! , (       d  f       g = f! [         R                  R                  R                  5          [         R                  R                  5          [         R                  R                  5       (       aO  [         R                  R                  SS9nUb/  [         R                  " UR                  5      R                  W5        [         R                  " W 5        S S S 5        O! , (       d  f       O= fS S S 5        f ! , (       d  f       f = f= f7f)NT)check_available)r   utils_mode_utilsno_dispatch_C_DisableFuncTorchget_rng_stateacceleratoris_availablecurrent_acceleratorget_device_moduler   set_rng_state)	rng_stater   accelerator_rng_states      r   r   r      s     
	 	 	,	,	.0J0J0L'')	))++++??PT?UK&(-(?(?$$)-/ & 1M	.+ [[$$002EHH4N4N4P  --//#//CC$( D  *++K,<,<=KK- 	* 5Q22' 1M0L	.	.& 5Q4P22U[[$$002EHH4N4N4P  --//#//CC$( D  *++K,<,<=KK- 	* 5Q4P4P222s   )LG
BGGL"H &(LG;-B	G*6G;>	L
G	G
G'"L*
G8	4G;;
H	L)L5L B	K&		L &
K40L 7	L 
L
LL)
contextlibr   r   contextmanagerr        r   <module>r(      s*       !+ !+r'   