
    Ǆg                     >    d dl Z d dlZd Ze j                  d        Zy)    Nc                 "   t               5  t        j                  d        | |i |}t        |t        j                        r7|j
                  j                  dk(  rt        j                  j                          |cddd       S # 1 sw Y   yxY w)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       \/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/torch/testing/_utils.pywrapper_set_seedr      sp     
	 	"T$V$fell+0B0Bf0L KK!!#	 	 	s   A0BBc            	   #     K   t         j                  j                  j                         5  t         j                  j                         5  t        j                         } t         j                  j                         rt         j                  j                         }d d d        d d d        	 d  t         j                  j                  j                         5  t         j                  j                         5  t         j                  j                         rt         j                  j                         t        j                          d d d        d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   *xY w# 1 sw Y   y xY w# t         j                  j                  j                         5  t         j                  j                         5  t         j                  j                         rt         j                  j                         t        j                          d d d        n# 1 sw Y   nxY wd d d        w # 1 sw Y   w xY wxY ww)N)
r   utils_mode_utilsno_dispatch_C_DisableFuncTorchget_rng_statecudais_availableset_rng_state)	rng_statecuda_rng_states     r   r   r      s     
	 	 	,	,	. 80J0J0L 8'')	::""$"ZZ557N8 8+ [[$$002 	+EHH4N4N4P 	+zz&&(

((8	*	+ 	+ 	+8 8 8 8	+ 	+ 	+ 	+U[[$$002 	+EHH4N4N4P 	+zz&&(

((8	*	+ 	+ 	+ 	+ 	+ 	+s   )IE'
AEE'#I,F 0(IE?7AE3
E?	IE$	 E''E0,I3E<	8E??FI)I4IAH/	&	I/H84I;	II	II)
contextlibr   r   contextmanagerr        r   <module>r%      s*       + +r$   