o hz@s<ddlZddlmZmZddlZGdddZdddZdS)N)AnyDictc@s4eZdZUeZeeefed<e Z ddZ dS) DeviceContext _CONTEXTScCs ||_dSN)device)selfrr P/var/www/html/ai/venv/lib/python3.10/site-packages/torch/_lazy/device_context.py__init__ s zDeviceContext.__init__N) __name__ __module__ __qualname__dictrrstrr__annotations__ threadingLock_CONTEXTS_LOCKr r r r r rs  rcCst|dur tjj}nt|}tjtj|d}|dur't|}|tj|<|WdS1s3wYdSr) torch_C_lazy_get_default_device_typerrrrget)rdevctxr r r get_device_contexts $rr)rtypingrrtorch._C._lazyrrrr r r r s