jHddlZddlmZmZddlZGddZddZdS)N)AnyDictcReZdZUiZeeefed<ej Z dZ dS) DeviceContext _CONTEXTSc||_dSN)device)selfr s KC:\Projects\Historical\venv\Lib\site-packages\torch/_lazy/device_context.py__init__zDeviceContext.__init__ s  N) __name__ __module__ __qualname__rrstrr__annotations__ threadingLock_CONTEXTS_LOCKr rr rrsL "ItCH~"""#Y^%%Nrrc<|$tjj}nt |}t j5t j|d}|t |}|t j|<|cdddS#1swxYwYdSr ) torch_C_lazy_get_default_device_typerrrrget)r devctxs r get_device_contextrs ~88::V  %(,,VT:: >"6**F.4M #F + sABBBr )rtypingrrtorch._C._lazyrrrrrr r"s{      r