o Ù,Ëh}ã@s’UddlZddlmZddlZddlmZddlmZmZm Z ddl m Z da eej ed<e d¡dd „ƒZGd d „d e ƒZd d „Zdd„ZdS)éN)ÚOptional)Ú_len_torch_function_stack)Ú _pop_modeÚ _push_modeÚTorchFunctionMode©Úcontext_decoratorÚCURRENT_DEVICEécCsâhtj’tj’tj’tj’tj’tj’tj’tj’tj ’tj j ’tj j ’tj ’tj’tj’tj’tj’tj’tjj’tj’tj’tj’tj’tj’tj’tj’tj’tj’tj’tj’tj ’tj!’tj"’tj#’tj$’tj%’tj&’S©N)'ÚtorchÚemptyÚempty_permutedÚ empty_stridedÚempty_quantizedÚonesÚarangeÚbartlett_windowÚblackman_windowÚeyeÚfftÚfftfreqÚrfftfreqÚfullÚhamming_windowÚ hann_windowÚ kaiser_windowÚlinspaceÚlogspaceÚnestedÚ nested_tensorÚrandÚrandnÚrandintÚrandpermÚrangeÚsparse_coo_tensorÚsparse_compressed_tensorÚsparse_csr_tensorÚsparse_csc_tensorÚsparse_bsr_tensorÚsparse_bsc_tensorÚ tril_indicesÚ triu_indicesÚzerosÚasarrayÚtensorÚ as_tensorÚ scalar_tensor©r3r3úN/var/www/html/scripts/venv/lib/python3.10/site-packages/torch/utils/_device.pyÚ_device_constructorss’þýüûúùø ÷ ö õ ô óòñðïîíêéèçæåäãâá à!ß"Þ#Ý$Ü&Ú'Ù(Ør5c@s.eZdZdd„Zdd„Zdd„Zd d d „ZdS) Ú DeviceContextcCst |¡|_dSr )r Údevice)Úselfr7r3r3r4Ú__init__>szDeviceContext.__init__cCsBt|_|jadd„ttƒƒDƒ}t|ƒt|ƒD]}t|ƒqdS)NcSsg|]}tƒ‘qSr3)r)Ú.0Ú_r3r3r4Ú Isz+DeviceContext.__enter__..)r Ú old_devicer7r%rrÚreversed)r8Ú cur_stackÚmoder3r3r4Ú __enter__As  ÿzDeviceContext.__enter__cCst|jag}ttƒdƒD]}tƒ}t|tƒrJ‚| |¡q tƒdkr-tƒ}t|tƒs-J‚t|ƒD]}t |ƒq1dS)Nr r) r=r r%rrÚ isinstancer6Úappendr>r)r8Úexc_typeÚexc_valÚexc_tbr?r;r@r3r3r4Ú__exit__Ps    ÿzDeviceContext.__exit__r3NcCs8|pi}|tƒvr| d¡dur|j|d<||i|¤ŽS)Nr7)r5Úgetr7)r8ÚfuncÚtypesÚargsÚkwargsr3r3r4Ú__torch_function__cs z DeviceContext.__torch_function__)r3N)Ú__name__Ú __module__Ú __qualname__r9rArGrMr3r3r3r4r6=s r6cst‡fdd„|ƒS)NcsˆSr r3r3©r7r3r4Úlsz"device_decorator..r)r7rIr3rQr4Údevice_decoratorksrScs ‡fdd„S)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)``. cstt ˆ¡|ƒSr )rSr r7)rIrQr3r4rRvszset_device..r3rQr3rQr4Ú set_deviceos rT)Ú functoolsÚtypingrr Útorch._CrÚtorch.overridesrrrÚtorch.utils._contextlibrr r7Ú__annotations__Ú lru_cacher5r6rSrTr3r3r3r4Ús     ..