`i^ddlZddlmZmZddlZGddZddeedefdZdS) N)AnyOptionalc\eZdZUiZeeefed<ej Z deddfdZ dS) DeviceContext _CONTEXTSdevicereturnNc||_dSN)r)selfrs n/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/_lazy/device_context.py__init__zDeviceContext.__init__ s  ) __name__ __module__ __qualname__rdictstrr__annotations__ threadingLock_CONTEXTS_LOCKrrr rrs_ "ItCH~"""#Y^%%Nstrrrr c<|$tjj}nt |}t j5t j|d}|t |}|t j|<|cdddS#1swxYwYdSr ) torch_C_lazy_get_default_device_typerrrrget)rdevctxs r get_device_contextr!s ~88::V  %(,,VT:: >"6**F.4M #F + sABBBr ) rtypingrrtorch._C._lazyrrrr!rrr r$s   x}       r