[iUddlZddlZddlmZddlmZmZmZddlm Z da ej dze d<ej ddZGdd eZd Zd ZdS) N)_len_torch_function_stack) _pop_mode _push_modeTorchFunctionModecontext_decoratorCURRENT_DEVICEchtjtjtjtjtjtjtjtjtj tj j tj j tj tjtjtjtjtjtjjtjtjtjtjtjtjtjtjtjtjtjtj tj!tj"tj#tj$tj%tj&SN)'torchemptyempty_permuted empty_stridedempty_quantizedonesarangebartlett_windowblackman_windoweyefftfftfreqrfftfreqfullhamming_window hann_window kaiser_windowlinspacelogspacenested nested_tensorrandrandnrandintrandpermrangesparse_coo_tensorsparse_compressed_tensorsparse_csr_tensorsparse_csc_tensorsparse_bsr_tensorsparse_bsc_tensor tril_indices triu_indiceszerosasarraytensor as_tensor scalar_tensorF/var/www/icac/venv/lib/python3.11/site-packages/torch/utils/_device.py_device_constructorsr7 s)  )  )   )   )   )   )  )  )   )   )   )   )  )  )  !) " #) $ %) &  "') ,  -) .  /) 0  1) 2 3) 4  5) 6 7) 8 &9) : ;) < =) > ?) @ A) B C) D E) F  G) H  I) L  M) N O) P Q) )r5c*eZdZddZdZdZd dZdS) DeviceContextreturnNc8tj||_dSr )r device)selfr<s r6__init__zDeviceContext.__init__=sl6** r5ct|_|jadtt D}t |t |D]}t |dS)Nc*g|]}tSr4)r).0_s r6 z+DeviceContext.__enter__..IsMMMQY[[MMMr5)r old_devicer<r&rrreversed)r= cur_stackmodes r6 __enter__zDeviceContext.__enter__Asu( NM%0I0K0K*L*LMMM 4Y''  D t      r5c|jag}ttdz D]I}t }t |t rtd||Jtdkr2t }t |t stdt|D]}t|dS)Nr z@Found nested DeviceContext on the mode stack where none expectedrz8Expected a DeviceContext at the bottom of the mode stack) rDr r&rr isinstancer9AssertionErrorappendrEr)r=exc_typeexc_valexc_tbrFrBrGs r6__exit__zDeviceContext.__exit__Ps 022Q677 # #A;;D$ .. $V   T " " " " $ & & * *;;DdM22 $NY''  D t      r5r4cx|pi}|tvr|d |j|d<||i|S)Nr<)r7getr<)r=functypesargskwargss r6__torch_function__z DeviceContext.__torch_function__isO2 ')) ) )fjj.B.B.J#{F8 tT$V$$$r5)r:N)r4N)__name__ __module__ __qualname__r>rHrPrWr4r5r6r9r9<sZ++++   2%%%%%%r5r9c*tfd|S)NcSr r4r<sr6z"device_decorator..rsVr5r)r<rSs` r6device_decoratorr_qs ^^^^T 2 22r5cfdS)z Set the default device inside of the wrapped function by decorating it with this function. If you would like to use this as a context manager, use device as a context manager directly, e.g., ``with torch.device(device)``. cHttj|Sr )r_r r<)rSr<s r6r^zset_device..|s(f)=)=tDDr5r4r]s`r6 set_devicerbus E D D DDr5) functoolsr torch._Crtorch.overridesrrrtorch.utils._contextlibrr r<__annotations__ lru_cacher7r9r_rbr4r5r6ris ......DDDDDDDDDD555555'+ t#***Q***\1%1%1%1%1%%1%1%1%j333EEEEEr5