3
Eg @ sr d dl Z d dlZd dlZG dd deZG dd deZe add Zdd ZG d
d d eZ G d d
d
e Z
dS ) Nc @ s: e Zd Zdd Zedd Zdd Zdd Zd
d
d Zd S ) _ContextInfoc C s || _ || _tj | _d S )N)cls
allow_default threadinglocal _local_stack)selfr r r A/usr/local/lib64/python3.6/site-packages/caffe2/python/context.py__init__
s z_ContextInfo.__init__c C s t | jdsg | j_| jjS )Nobj)hasattrr r )r r r r
_stack s z_ContextInfo._stackc C s | j j| d S )N)r append)r valuer r r
enter s z_ContextInfo.enterc C s2 t | jdks td| j | jj |ks.td S )Nr zContext %s is empty.)lenr AssertionErrorr pop)r r r r r
exit s z_ContextInfo.exitTc C sB t | jdkr8|sd S | js*td| j | j| j | jd S )Nr z*Context %s is required but none is active. )r r r r r r )r requiredr r r
get_active s z_ContextInfo.get_activeN)T) __name__
__module__ __qualname__r propertyr r r r r r r r
r s
r c @ s e Zd Zdd Zdd ZdS )_ContextRegistryc C s
i | _ d S )N)_ctxs)r r r r
r ' s z_ContextRegistry.__init__c C sD || j kr:t|ts"tdj|t|t|td| j |< | j | S )Nz'must be a context managed class, got {})r )r
issubclassManagedr formatr DefaultManaged)r r r r r
get* s
z_ContextRegistry.getN)r r r r r$ r r r r
r & s r c C s t S )N)_CONTEXT_REGISTRYr r r r
_context_registry4 s r& c C s dd t j| jD S )Nc S s* g | ]"}t |tr|tkr|tkr|qS r )r r! r# ).0r r r r