o Zh@sJddlZddlmZmZddlZGdddZd deedefddZdS) N)AnyOptionalc@s<eZdZUiZeeefed<e Z deddfddZ dS) DeviceContext _CONTEXTSdevicereturnNcCs ||_dSN)r)selfrr I/var/www/auris/lib/python3.10/site-packages/torch/_lazy/device_context.py__init__ s zDeviceContext.__init__) __name__ __module__ __qualname__rdictstrr__annotations__ threadingLock_CONTEXTS_LOCKr r r r r rs rrrcCst|dur tjj}nt|}tjtj|d}|dur't|}|tj|<|WdS1s3wYdSr) torchZ_CZ_lazyZ_get_default_device_typerrrrget)rZdevctxr r r get_device_contexts $rr) rtypingrrZtorch._C._lazyrrrrr r r r s