o ¡…›iœã@s†UddlZddlZddlmZddlmZmZmZddlm Z da ej dBe d<e  d¡dd„ƒZGd d „d eƒZd d „Zd d„ZdS)éN)Ú_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©r2r2úU/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/torch/utils/_device.pyÚ_device_constructors s’þýüûúùø ÷ ö õ ô óòñðïîíêéèçæåäãâá à!ß"Þ#Ý$Ü&Ú'Ù(Ør4c@s0eZdZd dd„Zdd„Zdd„Zd d d „ZdS)Ú DeviceContextÚreturnNcCst |¡|_dSr )r Údevice)Úselfr7r2r2r3Ú__init__=szDeviceContext.__init__cCsBt|_|jadd„ttƒƒDƒ}t|ƒt|ƒD]}t|ƒqdS)NcSsg|]}tƒ‘qSr2)r)Ú.0Ú_r2r2r3Ú Isz+DeviceContext.__enter__..)rÚ old_devicer7r$rrÚreversed)r8Ú cur_stackÚmoder2r2r3Ú __enter__As  ÿzDeviceContext.__enter__cCs||jag}ttƒdƒD]}tƒ}t|tƒrtdƒ‚| |¡q tƒdkr1tƒ}t|tƒs1tdƒ‚t |ƒD]}t |ƒq5dS)Nr z@Found nested DeviceContext on the mode stack where none expectedrz8Expected a DeviceContext at the bottom of the mode stack) r=rr$rrÚ isinstancer5ÚAssertionErrorÚappendr>r)r8Úexc_typeÚexc_valÚexc_tbr?r;r@r2r2r3Ú__exit__Ps$ ÿ   ÿ  ÿzDeviceContext.__exit__r2cCs8|pi}|tƒvr| d¡dur|j|d<||i|¤ŽS)Nr7)r4Úgetr7)r8ÚfuncÚtypesÚargsÚkwargsr2r2r3Ú__torch_function__is z DeviceContext.__torch_function__)r6N)r2N)Ú__name__Ú __module__Ú __qualname__r9rArHrNr2r2r2r3r5<s  r5cst‡fdd„|ƒS)NcsˆSr r2r2©r7r2r3Úrsz"device_decorator..r)r7rJr2rRr3Údevice_decoratorqsrTcs ‡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 )rTr r7)rJrRr2r3rS|szset_device..r2rRr2rRr3Ú set_deviceus rU)Ú functoolsr Útorch._CrÚtorch.overridesrrrÚtorch.utils._contextlibrrr7Ú__annotations__Ú lru_cacher4r5rTrUr2r2r2r3Ús    .5