U yh@s<ddlZddlmZmZddlZGdddZdddZdS)N)AnyDictc@s4eZdZUeZeeefed<e Z ddZ dS) DeviceContext _CONTEXTScCs ||_dSN)device)selfrr L/var/www/html/venv/lib/python3.8/site-packages/torch/_lazy/device_context.py__init__ szDeviceContext.__init__N) __name__ __module__ __qualname__dictrrstrr__annotations__ threadingLock_CONTEXTS_LOCKr r r r r rs rc Csh|dkrtjj}nt|}tj:tj|d}|dkrNt|}|tj|<|W5QRSQRXdSr) torchZ_CZ_lazyZ_get_default_device_typerrrrget)rZdevctxr r r get_device_contexts r)N)rtypingrrZtorch._C._lazyrrrr r r r s