Yh}UddlZddlmZddlZddlmZddlmZmZm Z ddl m Z da eej ed<ejddZGd d e Zd Zd ZdS) N)Optional)_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_tensore/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torch/utils/_device.py_device_constructorsr8s)  )  )   )   )   )   )  )  )   )   )   )   )  )  )  !) " #) $ %) &  "') ,  -) .  /) 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) )r6c(eZdZdZdZdZddZdS) DeviceContextc8tj||_dSr )rdevice)selfr<s r7__init__zDeviceContext.__init__>sl6** r6ct|_|jadtt D}t |t |D]}t |dS)Nc*g|]}tSr5)r).0_s r7 z+DeviceContext.__enter__..IsMMMQY[[MMMr6)r old_devicer<r'rrreversed)r= cur_stackmodes r7 __enter__zDeviceContext.__enter__Asu( NM%0I0K0K*L*LMMM 4Y''  D t      r6c~|jag}ttdz D]<}t }t |t rJ||=tdkr%t }t |t sJt|D]}t|dS)Nr r) rDr r'rr isinstancer:appendrEr)r=exc_typeexc_valexc_tbrFrBrGs r7__exit__zDeviceContext.__exit__Ps 022Q677 # #A;;D!$ 66 6 6 6   T " " " " $ & & * *;;DdM22 2 2 2Y''  D t      r6r5Ncx|pi}|tvr|d |j|d<||i|S)Nr<)r8getr<)r=functypesargskwargss r7__torch_function__z DeviceContext.__torch_function__csO2 ')) ) )fjj.B.B.J#{F8 tT$V$$$r6)r5N)__name__ __module__ __qualname__r>rHrOrVr5r6r7r:r:=sU+++   &%%%%%%r6r:c*tfd|S)NcSr r5r<sr7z"device_decorator..lsVr6r)r<rRs` r7device_decoratorr^ks ^^^^T 2 22r6cfdS)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^rr<)rRr<s r7r]zset_device..vs(f)=)=tDDr6r5r\s`r7 set_deviceraos E D D DDr6) functoolstypingrrtorch._Crtorch.overridesrrrtorch.utils._contextlibr r r<__annotations__ lru_cacher8r:r^rar5r6r7ris  ......DDDDDDDDDD555555*.&---Q***\*%*%*%*%*%%*%*%*%\333EEEEEr6