a h@sHddlZddlmZmZddlZGdddZdeeedddZdS) N)AnyOptionalc@s:eZdZUiZeeefed<e Z eddddZ dS) DeviceContext _CONTEXTSNdevicereturncCs ||_dSN)r)selfrr H/var/www/auris/lib/python3.9/site-packages/torch/_lazy/device_context.py__init__ szDeviceContext.__init__) __name__ __module__ __qualname__rdictstrr__annotations__ threadingLock_CONTEXTS_LOCKr r r r r rs rrcCst|durtjj}nt|}tj<tj|d}|durNt|}|tj|<|WdS1sf0YdSr ) torchZ_CZ_lazyZ_get_default_device_typerrrrget)rZdevctxr r r get_device_contexts r)N) rtypingrrZtorch._C._lazyrrrrr r r r s