o wZŽhÍã@s’UddlmZddlZddlmZmZmZddlmZddl m Z ddl Z da eej ed<e  d¡dd „ƒZGd d „d eƒZd d „Zdd„ZdS)é)ÚOptionalN)ÚTorchFunctionModeÚ _pop_modeÚ _push_mode©Úcontext_decorator)Ú_len_torch_function_stackÚCURRENT_DEVICEécCsúhtj’tj’tj’tj’tj’tj’tj’tj’tj ’tj j ’tj j ’tj ’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&’tj'’tj(’tj%’S©N))ÚtorchÚemptyZempty_permutedZ empty_stridedZempty_quantizedZonesZarangeZbartlett_windowZblackman_windowÚeyeZfftZfftfreqZrfftfreqÚfullÚfillZhamming_windowZ hann_windowZ kaiser_windowZlinspaceZlogspaceÚnestedZ nested_tensorZrandZrandnÚrandintZrandpermÚrangeZsparse_coo_tensorZsparse_compressed_tensorZsparse_csr_tensorZsparse_csc_tensorZsparse_bsr_tensorZsparse_bsc_tensorZ tril_indicesZ triu_indicesZvanderZzerosZasarrayZtensorZ as_tensorZ scalar_tensor©rrúB/var/www/auris/lib/python3.10/site-packages/torch/utils/_device.pyÚ_device_constructors s¢þýüûúùø ÷ ö õ ô óòñðïîíìéèçæåäãâá à!ß"Þ#Ý$Ü%Û&Ú'Ù)×*Ö+Õ,Ôrc@s.eZdZdd„Zdd„Zdd„Zd d d „ZdS) Ú DeviceContextcCst |¡|_dSr )r Údevice)ÚselfrrrrÚ__init__>szDeviceContext.__init__cCsBt|_|jadd„ttƒƒDƒ}t|ƒt|ƒD]}t|ƒqdS)NcSsg|]}tƒ‘qSr)r)Ú.0Ú_rrrÚ Isz+DeviceContext.__enter__..)r Ú old_devicerrrrÚreversed)rÚ cur_stackÚmoderrrÚ __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) rr rrrÚ isinstancerÚappendrr)rÚexc_typeÚexc_valÚexc_tbr rr!rrrÚ__exit__Qs    ÿzDeviceContext.__exit__rNcCs8|pi}|tƒvr| d¡dur|j|d<||i|¤ŽS)Nr)rÚgetr)rÚfuncÚtypesÚargsÚkwargsrrrÚ__torch_function__ds z DeviceContext.__torch_function__)rN)Ú__name__Ú __module__Ú __qualname__rr"r(r.rrrrr=s rcst‡fdd„|ƒS)NcsˆSr rr©rrrÚlsz"device_decorator..r)rr*rr2rÚdevice_decoratorksr4cs ‡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 )r4r r)r*r2rrr3uszset_device..rr2rr2rÚ set_devicens r5)Útypingrr Ztorch.overridesrrrZtorch.utils._contextlibrZtorch._CrÚ functoolsr rÚ__annotations__Ú lru_cacherrr4r5rrrrÚs   1.