U Mh @s~UddlmZddlZddlmZddlmZddlZdaeej e d<e dddZ Gd d d eZ d d Zd dZdS))OptionalN)TorchFunctionModecontext_decoratorCURRENT_DEVICEc(Cstjtjtjtjtjtjtjtjtj tj j tj j tj tjtjtjtjtjtjtjjtjtjtjtjtjtjtjtjtjtjtjtj tj!tj"tj#tj$tj%tj&tj'tj(tj%h(SN))torchemptyZempty_permutedZ empty_stridedZempty_quantizedZonesZarangeZbartlett_windowZblackman_windoweyeZfftZfftfreqZrfftfreqfullfillZhamming_windowZ hann_windowZ kaiser_windowZlinspaceZlogspacenestedZ nested_tensorZrandZrandnrandintZrandpermrangeZsparse_coo_tensorZsparse_compressed_tensorZsparse_csr_tensorZsparse_csc_tensorZsparse_bsr_tensorZsparse_bsc_tensorZ tril_indicesZ triu_indicesZvanderZzerosZasarrayZtensorZ as_tensorZ scalar_tensorrrE/var/www/html/venv/lib/python3.8/site-packages/torch/utils/_device.py_device_constructors sRrcs:eZdZddZfddZfddZd d d ZZS) DeviceContextcCst||_dSr)r device)selfrrrr__init__=szDeviceContext.__init__cst|_|jatSr)r old_devicersuper __enter__)r __class__rrr@szDeviceContext.__enter__cs|jat|||Sr)rrr__exit__)rexc_typeexc_valexc_tbrrrrFszDeviceContext.__exit__rNcCs4|pi}|tkr*|ddkr*|j|d<|||S)Nr)rgetr)rfunctypesargskwargsrrr__torch_function__Ks z DeviceContext.__torch_function__)rN)__name__ __module__ __qualname__rrrr& __classcell__rrrrr<s  rcstfdd|S)NcsSrrrrrrSz"device_decorator..r)rr"rr+rdevice_decoratorRsr.cs fddS)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)r.r r)r"r+rrr,\r-zset_device..rr+rr+r set_deviceUsr/)typingrr Ztorch.overridesrZtorch.utils._contextlibr functoolsrr__annotations__ lru_cacherrr.r/rrrrs    1