o j9:j@sFddlZddlmZddlZGdddZd dedBdefddZdS) N)Anyc@s<eZdZUiZeeefed<e Z deddfddZ dS) DeviceContext _CONTEXTSdevicereturnNcCs ||_dSN)r)selfrr a/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/torch/_lazy/device_context.py__init__ s zDeviceContext.__init__) __name__ __module__ __qualname__rdictstrr__annotations__ threadingLock_CONTEXTS_LOCKr r r r r rs rrrcCst|dur tjj}nt|}tjtj|d}|dur't|}|tj|<|WdS1s3wYdSr) torch_C_lazy_get_default_device_typerrrrget)rdevctxr r r get_device_contexts $rr)rtypingrtorch._C._lazyrrrrr r r r s