a ¢º”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éc$Csštjtjtjtjtjtjtjtjtj tj j tj j tj tjtjtjtjtjtjjtjtjtjtjtjtjtjtjtjtjtjtj tj!tj"tj#tj$tj%tj&h$S©N)'ÚtorchÚemptyZempty_permutedZ empty_stridedZempty_quantizedZonesZarangeZbartlett_windowZblackman_windowZeyeZfftZfftfreqZrfftfreqÚfullZhamming_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_indicesÚzerosZasarrayZtensorZ as_tensorZ scalar_tensor©rrúA/var/www/auris/lib/python3.9/site-packages/torch/utils/_device.pyÚ_device_constructorssJØ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|ƒq0dS)NcSsg|] }tƒ‘qSr)r)Ú.0Ú_rrrÚ Ióz+DeviceContext.__enter__..)r Ú old_devicerrrrÚreversed)rÚ cur_stackÚmoderrrÚ __enter__As  zDeviceContext.__enter__cCst|jag}ttƒdƒD]"}tƒ}t|tƒr0J‚| |¡qtƒdkrZtƒ}t|tƒsZJ‚t|ƒD] }t |ƒqbdS)Nr r) rr rrrÚ isinstancerÚappendrr)rÚexc_typeÚexc_valÚexc_tbr rr!rrrÚ__exit__Ps   zDeviceContext.__exit__rNcCs8|pi}|tƒvr*| d¡dur*|j|d<||i|¤ŽS)Nr)rÚgetr)rÚfuncÚtypesÚargsÚkwargsrrrÚ__torch_function__cs z DeviceContext.__torch_function__)rN)Ú__name__Ú __module__Ú __qualname__rr"r(r.rrrrr=srcst‡fdd„|ƒS)NcsˆSr rr©rrrÚlrz"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*r2rrr3vrzset_device..rr2rr2rÚ set_deviceosr5)Ú functoolsÚtypingrr Ztorch._CrZtorch.overridesrrrZtorch.utils._contextlibrr rÚ__annotations__Ú lru_cacherrr4r5rrrrÚs     ..