o ,’×i× ã@s~UddlmZddlZddlmZddlmZddlZdaeej e d<e  d¡dd„ƒZ Gd d „d eƒZ d d „Zd d„ZdS)é)ÚOptionalN)Ú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’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ÚemptyÚempty_permutedÚ empty_stridedÚempty_quantizedÚonesÚarangeÚbartlett_windowÚblackman_windowÚeyeÚfftÚfftfreqÚrfftfreqÚfullÚfillÚ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ÚvanderÚzerosÚasarrayÚtensorÚ as_tensorÚ scalar_tensor©r2r2úQ/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torch/utils/_device.pyÚ_device_constructors s¢þýüûúùø ÷ ö õ ô óòñðïîíìéèçæåäãâá à!ß"Þ#Ý$Ü%Û&Ú'Ù)×*Ö+Õ,Ôr4cs:eZdZdd„Z‡fdd„Z‡fdd„Zd d d „Z‡ZS) Ú DeviceContextcCst |¡|_dSr)r Údevice)Úselfr6r2r2r3Ú__init__=szDeviceContext.__init__cst|_|jatƒ ¡Sr)rÚ old_devicer6ÚsuperÚ __enter__)r7©Ú __class__r2r3r;@s zDeviceContext.__enter__cs|jatƒ |||¡Sr)r9rr:Ú__exit__)r7Úexc_typeÚexc_valÚexc_tbr<r2r3r>FszDeviceContext.__exit__r2NcCs8|pi}|tƒvr| d¡dur|j|d<||i|¤ŽS)Nr6)r4Úgetr6)r7ÚfuncÚtypesÚargsÚkwargsr2r2r3Ú__torch_function__Ks z DeviceContext.__torch_function__)r2N)Ú__name__Ú __module__Ú __qualname__r8r;r>rGÚ __classcell__r2r2r<r3r5<s   r5cst‡fdd„|ƒS)NcsˆSrr2r2©r6r2r3ÚSsz"device_decorator..r)r6rCr2rLr3Údevice_decoratorRsrNcs ‡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)rNr r6)rCrLr2r3rM\szset_device..r2rLr2rLr3Ú set_deviceUs rO)Útypingrr Útorch.overridesrÚtorch.utils._contextlibrÚ functoolsrr6Ú__annotations__Ú lru_cacher4r5rNrOr2r2r2r3Ús   1