U
    ªÊ¹jÖ‡  ã                   @   sª  U d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm	Z	m
Z
mZ d dlZd dlmZ d dlmZ eedƒo|eedƒZe jdd„ ƒZG d	d
„ d
ƒZejjZejjZeedœdd„Zi Zi ZG dd„ deƒZdd„ Zdd„ Zdd„ Z dd„ Z!i a"eejje	f e#d< e jdd„ ƒZ$dd„ Z%dd„ Z&dd „ Z'e(ƒ a)d!d"„ Z*d#d$„ Z+d%d&„ Z,G d'd(„ d(eƒZ-G d)d*„ d*ƒZ.G d+d,„ d,ej/ƒZ0G d-d.„ d.e0ƒZ1G d/d0„ d0ej/ƒZ2e2ƒ Z3dS )1é    N)ÚAnyÚCallableÚDictÚListÚTypeÚUnion)Ú_utils_internal)Údispatch_functorchÚgetdlopenflagsÚsetdlopenflagsc               	   c   sB   t sdV  dS t ¡ } t | tjB ¡ z
dV  W 5 t | ¡ X dS )z�
    Context manager to set the RTLD_GLOBAL dynamic linker flag while we open a
    shared library to load custom operators.
    N)Ú_SET_GLOBAL_FLAGSÚsysr
   r   ÚctypesÚRTLD_GLOBAL)Z	old_flags© r   úI/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/_ops.pyÚdl_open_guard   s    
r   c                   @   s@   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dS )ÚOperatorBasez·
    Base class for OpOverload (which represents C++ ATen operators) and HigherOrderOperator
    (which represents Python-only operators that are unrepresentable in TorchScript).
    c                 C   s(   i | _ i | _ddlm} i | _i | _d S )Nr   )ÚTorchDispatchMode)Ú_dispatch_cacheÚ
py_kernelsÚtorch.utils._python_dispatchr   Úpython_key_mode_tableÚfunctorch_table)Úselfr   r   r   r   Ú__init__)   s    þ
þzOperatorBase.__init__c                 O   s
   t ƒ ‚d S ©N©ÚNotImplementedError©r   ÚargsÚkwargsr   r   r   Ú__call__W   s    zOperatorBase.__call__c                 C   s
   || j kS r   )r   ©r   Úkr   r   r   Úhas_kernel_for_dispatch_keyZ   s    z(OperatorBase.has_kernel_for_dispatch_keyc                 C   s,   | j D ] }tj |¡s| |¡r dS qdS )NTF)r   ÚtorchÚ_CZ_dispatch_is_alias_keyÚhas)r   Úksr$   r   r   r   Úhas_kernel_for_any_dispatch_key]   s    
z,OperatorBase.has_kernel_for_any_dispatch_keyc                    s   ‡ ‡fdd„}|S )Nc                    sÎ   t  ˆ ¡r@tˆ tjjjƒr@ˆ ˆjks(t‚| ˆjˆ < ˆj	 
¡  | S tˆ tjjjƒrlˆ ˆjks^t‚| ˆjˆ < | S tˆ tjjƒs~t‚ˆ tjjjks”tdƒ‚ˆ ˆjkr¶tdˆ › dˆ ¡ › �ƒ‚| ˆjˆ < ˆj	 
¡  | S )NzGPlease register a mode for the torch._C.DispatchKey.Python key instead.z%Trying to override a python impl for z on operator )ÚinspectÚisclassÚ
issubclassr&   ÚutilsÚ_python_dispatchr   r   ÚAssertionErrorr   ÚclearÚ
isinstancer'   Z
_functorchZTransformTyper   ÚDispatchKeyÚPythonr   ÚRuntimeErrorÚname)Úfn©r$   r   r   r   Úinnerd   s0     ÿ


ÿþ
ÿ

z#OperatorBase.py_impl.<locals>.innerr   )r   r$   r9   r   r8   r   Úpy_implc   s    zOperatorBase.py_implc                 C   s
   t ƒ ‚d S r   r   ©r   r   r   r   r6   ‚   s    zOperatorBase.nameN)
Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r"   r%   r*   r:   r6   r   r   r   r   r   #   s   .r   )Úopr$   c                 C   sd  |   |¡r|S tj}|tjks(t||ƒr6|   |¡r6|S tj}|tjksPt||ƒr^|   |¡r^|S |  tj 	|¡¡pz|   tj¡}tj
}|tjkr¨t||ƒr¨|   |¡r¨|s¨|S tj}|tjksÂt||ƒrö|   |¡rö|tjkrî|  tjj¡rîtdƒ‚n|sö|S tj}t||ƒ�r|   |¡�r|S tj}t||ƒ�r:|   |¡�r:|S tj |¡�rL|S td| › d|› �ƒ‚d S )Nzambiguous autogradother kernelzcould not find kernel for z at dispatch key )r%   r3   Z&CompositeExplicitAutogradNonFunctionalZ	UndefinedÚis_included_in_aliasZCompositeExplicitAutogradr*   r&   r'   Z*_dispatch_get_backend_keyset_from_autogradZ%CompositeImplicitAutogradNestedTensorÚCompositeImplicitAutogradZAutogradOtherZ _dispatch_autogradother_backendsr5   ZAutogradZFuncTorchBatchedDecompositionZ_dispatch_has_backend_fallbackr   )r@   r$   ÚcandZhas_backend_kernelr   r   r   Úresolve_keyŒ   sn    
ÿÿþÿÿþ
ÿ
þÿÿþýÿÿþÿ
rD   c                       sV   e Zd Zddœ‡ fdd„
Zedd„ ƒZdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
‡  ZS )ÚHigherOrderOperatorF)Ú_deprecated_global_nsc                   sr   t ƒ  ¡  || _|| _|r*| t|< d | _n| t|< d| _| jtkrb| j	rRd| j	 nd}| j
| | _
tj ¡ | _d S )NÚhigher_orderÚ.Ú )Úsuperr   Ú_namer<   Ú_global_higher_order_opsÚ_nsÚ_higher_order_opsÚ	__class__rE   Ú	namespacer=   r&   r'   Z_dispatch_keyset_fullÚnon_fallthrough_keys)r   r6   rF   Zself_name_space©rO   r   r   r   Í   s    

zHigherOrderOperator.__init__c                 C   s   | j S r   )rM   r;   r   r   r   rP   ã   s    zHigherOrderOperator.namespacec                 C   s   | j  |¡| _ d S r   )rQ   Úremove)r   Údispatch_keyr   r   r   Úfallthroughç   s    zHigherOrderOperator.fallthroughc           
      O   sh  ddl m} || jkr<| j| }t|tjjƒr2t‚|||ŽS |tjjjkrVt	| ||ƒS |tjjj
kr¬|ƒ }|d k	sztdƒ‚t|ƒ| jks˜td|› d�ƒ‚| jt|ƒ ||ŽS tj |¡}|tƒ k�rtƒ | }t|ƒdk�rtj tj
¡�s|d }| jt|ƒ ||ŽS t| |ƒ}	|	| jk�s0t|› d|	› �ƒ‚| j|	 | j|< | j|	 }t|tjjƒ�r^t‚|||ŽS )Nr   ©Ú_get_current_dispatch_modeúMIllegal invocation of dispatch on torch._C.DispatchKey.Python without a mode.zCurrent active mode z not registeredéÿÿÿÿz -> )r   rW   r   r2   r&   r'   r3   r0   ZFuncTorchDynamicLayerFrontModer	   r4   Útyper   Ú_to_functionality_keyÚmode_stack_per_keyÚlenÚ&_dispatch_tls_is_dispatch_key_excludedrD   r   )
r   rT   r    r!   rW   ZkernelÚ	curr_modeÚfunctionality_keyÚ
curr_stackÚ	final_keyr   r   r   Údispatchê   sL    


ÿþÿ
þ
ÿþÿþ

zHigherOrderOperator.dispatchc                    s0   dd l ‰ddl m} |‡ ‡‡‡fdd„ƒ}|ƒ S )Nr   )Údisablec                     sP   t ˆ ˆƒ} ˆj | ¡r,ˆjjˆ| fˆ žˆŽS tˆ ˆˆjƒ}ˆj| ¡ fˆ žˆŽS r   )Ú_to_flat_tupleZ	overridesZhas_torch_functionZhandle_torch_functionÚ_compute_keysetrQ   rc   ZhighestPriorityTypeId)Ú	flat_argsZdispatch_key_set©r    r!   r   r&   r   r   Úwrapper   s$    
 ÿÿÿÿÿÿz-HigherOrderOperator.__call__.<locals>.wrapper)Ztorch._dynamord   )r   r    r!   rd   ri   r   rh   r   r"     s
    zHigherOrderOperator.__call__c                 C   s
   |   ¡ › S r   )r6   r;   r   r   r   Ú__str__/  s    zHigherOrderOperator.__str__c                 C   s   | j S r   ©rK   r;   r   r   r   r6   2  s    zHigherOrderOperator.name)r<   r=   r>   r   ÚpropertyrP   rU   rc   r"   rj   r6   Ú__classcell__r   r   rR   r   rE   Å   s   
0rE   c                 C   s0   t jj | ¡\}}t jj |¡\}}|| }|S r   )r&   r.   Ú_pytreeÚtree_flatten)r    r!   rg   Ú_Zflat_kwargsÚflat_allr   r   r   re   6  s    re   c                 C   s   t | |ƒ}t||ƒS r   )Ú_get_tensorsÚkey_extractor)r    r!   rQ   Útensorsr   r   r   rf   =  s    
rf   c                 C   s    t | |ƒ}dd„ |D ƒ}t|ƒS )Nc                 S   s   g | ]}t |tjƒr|‘qS r   )r2   r&   ÚTensor)Ú.0Útr   r   r   Ú
<listcomp>D  s      z _get_tensors.<locals>.<listcomp>)re   Útuple)r    r!   rq   Ztensor_argsr   r   r   rr   B  s    
rr   c                 C   s>   t j ¡ }| D ]}|t j |¡B }q|t j ¡  }||@ }|S r   )r&   r'   Z_dispatch_tls_local_include_setÚ_dispatch_keysZ_dispatch_tls_local_exclude_set)rt   Zkey_maskZkey_setZtensorr   r   r   rs   J  s    
rs   Ú_mode_stack_per_keyc              	   c   s4   t | ƒdkst‚|  ¡ }z
|V  W 5 |  |¡ X d S ©Nr   )r]   r0   ÚpopÚappend)Z
mode_stackZtop_moder   r   r   Útemporarily_pop_mode`  s
    
r   c                   C   s   t S r   )r{   r   r   r   r   r\   j  s    r\   c                 C   sN   t | tjjƒst‚t |tjjjƒs&t‚| tƒ kr:g tƒ | < tƒ |   	|¡ d S r   )
r2   r&   r'   r3   r0   r.   r/   r   r\   r~   )ÚkeyÚmoder   r   r   Úpush_mode_for_keyp  s
    

r‚   c                 C   sB   t | tjjƒst‚| tƒ ks t‚tƒ |  }t|ƒdks:t‚| ¡ S r|   )r2   r&   r'   r3   r0   r\   r]   r}   )r€   Zcurr_mode_stackr   r   r   Úpop_mode_for_keyy  s
    
rƒ   c                 C   s   t  | ¡ d S r   )Ú
cached_opsÚadd)Zop_overloadr   r   r   Úadd_cached_op„  s    r†   c                   C   s   t  ¡  d S r   )r„   r1   r   r   r   r   Úreset_cached_ops‰  s    r‡   c                   C   s   t S r   )r„   r   r   r   r   Úget_cached_opsŽ  s    rˆ   c                       s®   e Zd Z‡ fdd„Zd"dd„Zdd„ Zdd	„ Zd
d„ Zdd„ Z‡ fdd„Z	‡ fdd„Z
edd„ ƒZdd„ Zdd„ Zdd„ Zdd„ Zedd„ ƒZedd„ ƒZed d!„ ƒZ‡  ZS )#Ú
OpOverloadc                    sð   t ƒ  ¡  || _|| _|| _|| _|| _|jdkr6dn|j| _| jj	| _
|jrb|  j
d|j 7  _
| jj	 d¡d › d| j› �| _|j| _|j|_| j
| _i | _d }| jjD ].}|jd kr¼q¬|d krÎ|jj}q¬|jjpØ|}q¬|d k	oè| | _d S )NrI   ÚdefaultrH   ú::é   )rJ   r   Ú_opÚ_op_dkÚ_schemaÚ_overloadpacketÚ_tagsZoverload_nameÚ_overloadnamer6   rK   Úsplitr<   r=   r>   Ú__annotations__Ú	argumentsZ
alias_infoÚis_writeZis_view)r   Úoverloadpacketr@   Zop_dkÚschemaÚtagsr–   ÚarR   r   r   r   –  s0    
ÿ
 

zOpOverload.__init__Nc                 C   s   | S r   r   ©r   Úmemor   r   r   Ú__deepcopy__·  s    zOpOverload.__deepcopy__c                 C   s   dj | jj d¡| jfžŽ S )Nz'<OpOverload(op='{}.{}', overload='{}')>r‹   ©Úformatr�   r6   r“   r’   r;   r   r   r   Ú__repr__º  s     ÿzOpOverload.__repr__c                 O   s   | j ||pi ŽS r   ©r�   r   r   r   r   r"   ¿  s    zOpOverload.__call__c                 C   s
   t | jƒS r   ©Úhashr�   r;   r   r   r   Ú__hash__Â  s    zOpOverload.__hash__c                 C   s   dj | jj d¡| jfžŽ S )Nz{}.{}.{}r‹   rž   r;   r   r   r   rj   Æ  s    zOpOverload.__str__c                    s   t ƒ  |¡ptj |  ¡ |¡S r   )rJ   r%   r&   r'   Ú%_dispatch_has_kernel_for_dispatch_keyr6   r#   rR   r   r   r%   É  s
    ÿþz&OpOverload.has_kernel_for_dispatch_keyc                    s   t j |  ¡ |¡ptƒ  |¡S r   )r&   r'   Z)_dispatch_has_kernel_for_any_dispatch_keyr6   rJ   r*   )r   r)   rR   r   r   r*   Î  s     ÿ
þz*OpOverload.has_kernel_for_any_dispatch_keyc                 C   s   | j j d¡d S )Nr‹   r   )r�   r6   r“   r;   r   r   r   rP   Ó  s    zOpOverload.namespacec                 O   sP   t jjj}|| jkr$| j| ||ŽS t j |  ¡ |¡rH| j|f|ž|ŽS tS d S r   )	r&   r'   r3   rB   r   r¥   r6   rŽ   ÚNotImplemented)r   r    r!   Zdkr   r   r   Ú	decompose×  s    

zOpOverload.decomposec                 C   s   | j  |d ¡ d S r   )r   r}   )r   r€   r   r   r   Ú_uncache_dispatchì  s    zOpOverload._uncache_dispatchc                    sD  ˆˆj kstˆ› dˆ› �ƒ‚ˆtjjjkrjˆjsFˆˆj ˆ< tˆƒ ˆS ‡‡fdd„}|ˆj ˆ< tˆƒ |S d}tj ˆ¡}|t	ƒ kr¾t	ƒ | ‰ t
ˆ ƒdkrºtj tj¡sº‡ ‡fdd„}|S d}tˆˆƒ}ˆtjjjk�rdd lm  m} |j�r| ˆ|¡}|�r|ˆj ˆ< tˆƒ |S ˆj ||¡}|�r@|ˆj ˆ< tˆƒ |S )Nú c                     sR   ddl m} t|ƒ ƒ}|d k	s&tdƒ‚|ˆjkrBˆjˆ f| ž|ŽS ˆj| | |ŽS )Nr   rV   rX   )r   rW   rZ   r0   r   rŽ   )r    r!   rW   r_   )r€   r   r   r   Úhandlerú  s    
ÿþ
z)OpOverload._get_dispatch.<locals>.handlerTr   c               
      s˜   t ˆ ƒ�†}t|dƒst‚g }tjj | | ¡ f¡\}}|D ]6}t|tj	ƒr:tj
 |¡ tj
jj¡r:| t|ƒ¡ q:| ˆ|| |¡W  5 Q R £ S Q R X d S )NÚ__torch_dispatch__)r   Úhasattrr0   r&   r.   rn   ro   Úvaluesr2   ru   r'   rz   r(   r3   r4   r~   rZ   r«   )r    r!   r_   Zoverload_typesZargs_flattenedrp   rš   )ra   r   r   r   rª     s&    

ÿÿþ   ÿF)r   r0   r&   r'   r3   r4   r   r†   r[   r\   r]   r^   rD   ZFunctionalizeZtorch._dispatch.pythonZ	_dispatchÚpythonZCROSSREF_FUNCTIONALIZEZmake_crossref_functionalizer   Úget)r   r€   rª   Zcache_resultr`   rb   Z
pydispatchÚrr   )ra   r€   r   r   Ú_get_dispatchð  sP    



ÿþÿþ


zOpOverload._get_dispatchc                 C   s   | j S r   rk   r;   r   r   r   r6   P  s    zOpOverload.namec                 C   s   | j S r   )r�   r;   r   r   r   r—   S  s    zOpOverload.overloadpacketc                 C   s   | j S r   r¡   r;   r   r   r   r@   W  s    zOpOverload.opc                 C   s   | j S r   )r‘   r;   r   r   r   r™   [  s    zOpOverload.tags)N)r<   r=   r>   r   r�   r    r"   r¤   rj   r%   r*   rl   rP   r§   r¨   r±   r6   r—   r@   r™   rm   r   r   rR   r   r‰   •  s(   !

`

r‰   c                   @   sb   e Zd Zdd„ Zddd„Zdd„ Zdd	„ Zd
d„ Zedd„ ƒZ	dd„ Z
dd„ Zdd„ Zdd„ ZdS )ÚOpOverloadPacketc                 C   s"   || _ || _|| _|| _g | _d S r   )Ú_qualified_op_namer<   r�   Ú_overload_namesÚ_dir)r   Úqualified_op_nameÚop_namer@   Úoverload_namesr   r   r   r   e  s
    zOpOverloadPacket.__init__Nc                 C   s   | S r   r   r›   r   r   r   r�   o  s    zOpOverloadPacket.__deepcopy__c                 C   s   dj | j d¡Ž S )Nz<OpOverloadPacket(op='{}.{}')>r‹   ©rŸ   r³   r“   r;   r   r   r   r    r  s    
ÿzOpOverloadPacket.__repr__c                 C   s
   t | jƒS r   r¢   r;   r   r   r   r¤   w  s    zOpOverloadPacket.__hash__c                 C   s   dj | j d¡Ž S )Nz{}.{}r‹   r¹   r;   r   r   r   rj   z  s    zOpOverloadPacket.__str__c                 C   s   | j S r   r¡   r;   r   r   r   r@   }  s    zOpOverloadPacket.opc                 C   s   |dkrdS z|  d¡r&t| j|ƒW S W n< tk
rd   tdt| ƒ› dt| jƒ› d|› d�ƒd ‚Y nX zd|dkrtd	n|}tj | j|¡\}}}tj 	| j|¡}t
| ||||ƒ}t| ||ƒ | j |¡ |W S  tk
rú   td
t| ƒ› d|› d�ƒd ‚Y nX d S )NÚ__file__ú	torch.opsÚ__ú'zH' can't have an overload name beginning with '__' and the underlying op z has no attribute z either.rŠ   rI   zThe underlying op of 'z' has no overload name ')Ú
startswithÚgetattrr�   ÚAttributeErrorÚstrr&   r'   Z_get_operation_overloadr³   Z_get_schemar‰   Úsetattrrµ   r~   r5   )r   r€   Zuse_keyZop_Zop_dk_r™   r˜   Úoverloadr   r   r   Ú__getattr__�  s8    	
 ÿý ÿ
ÿþzOpOverloadPacket.__getattr__c                 C   s
   t | jƒS r   ©Úiterrµ   r;   r   r   r   Ú__iter__¬  s    zOpOverloadPacket.__iter__c                 O   s   | j ||pi ŽS r   r¡   r   r   r   r   r"   ¯  s    zOpOverloadPacket.__call__c                 C   s   dd„ | j D ƒS )Nc                 S   s   g | ]}|r|nd ‘qS )rŠ   r   )rv   Únr   r   r   rx   ¸  s     z.OpOverloadPacket.overloads.<locals>.<listcomp>)r´   r;   r   r   r   Ú	overloads·  s    zOpOverloadPacket.overloads)N)r<   r=   r>   r   r�   r    r¤   rj   rl   r@   rÄ   rÇ   r"   rÉ   r   r   r   r   r²   d  s   


+r²   c                       s0   e Zd ZdZ‡ fdd„Zdd„ Zdd„ Z‡  ZS )Ú_OpNamespacea0  
    An op namespace to dynamically bind Operators into Python.

    Say a user has created a custom Operator called "my_namespace::my_op". To
    call this op, the user will write torch.ops.my_namespace.my_op(...).
    At startup, this operation will not yet be bound into Python. Instead, the
    following sequence of magic tricks will occur:
    1. `torch.ops.my_namespace` will invoke the `__getattr__` magic method
       on the `torch.ops` object, which will create a new `_OpNamespace`
       object called `my_namespace` and set it as an attribute on the `ops`
       object.
    2. `torch.ops.my_namespace.my_op` will then invoke `__getattr__` on
       the `my_namespace` object, which will retrieve the operation via
       `torch.get_operation`, a function bound from C++, and then in a similar
       fashion bind this new object onto the `my_namespace` object.
    3. `torch.ops.my_namespace.my_op(...)` then calls this new operation
        and subsequent accesses will incur no further lookup (the namespace and
        operation will already exist).
    c                    s    t ƒ  d| ¡ || _g | _d S )Nz
torch.ops.)rJ   r   r6   rµ   )r   r6   rR   r   r   r   ß  s    z_OpNamespace.__init__c                 C   s
   t | jƒS r   rÅ   r;   r   r   r   rÇ   ä  s    z_OpNamespace.__iter__c              
   C   sì   |dkrdS |dkr,t d|› d| j› d�ƒ‚| j}|› d|› �}ztj |¡\}}W n< tk
r� } zt d| j› d	|› d�ƒ|‚W 5 d }~X Y nX tjj ||¡ | j	d
 | |_	t
||||ƒ}| j	d
 | |_	t| ||ƒ | j |¡ |S )Nrº   r»   )Ú
__origin__Ú__self__zInvalid attribute 'z' for '_OpNamespace' 'r½   r‹   z'_OpNamespace' 'ú' object has no attribute 'rH   )rÀ   r6   r&   r'   Z_jit_get_operationr5   ZjitZ	_builtinsZ_register_builtinr=   r²   rÂ   rµ   r~   )r   r·   Znamespace_namer¶   r@   r¸   ÚeZopoverloadpacketr   r   r   rÄ   ç  s8    ÿÿþ   ÿz_OpNamespace.__getattr__)r<   r=   r>   r?   r   rÇ   rÄ   rm   r   r   rR   r   rÊ   Ê  s   rÊ   c                       s$   e Zd Z‡ fdd„Zdd„ Z‡  ZS )Ú_PyOpNamespacec                    s   t ƒ  |¡ || _d S r   )rJ   r   Ú_ops)r   r6   ÚopsrR   r   r   r     s    z_PyOpNamespace.__init__c                 C   s>   | j  |d ¡}|d kr.td| j› d|› d�ƒ‚t| ||ƒ |S )Nz'_PyOpNamespace' 'rÍ   r½   )rÐ   r¯   rÀ   r6   rÂ   )r   r6   r@   r   r   r   rÄ     s    ÿz_PyOpNamespace.__getattr__)r<   r=   r>   r   rÄ   rm   r   r   rR   r   rÏ     s   rÏ   c                       s8   e Zd ZdZ‡ fdd„Zdd„ Zdd„ Zdd	„ Z‡  ZS )
Ú_Opsz_ops.pyc                    s6   t ƒ  d¡ tƒ | _tdtƒ| _tdtƒ| _g | _	d S )Nr»   ztorch.ops.higher_order)
rJ   r   ÚsetÚloaded_librariesrÏ   rL   Ú!_global_higher_order_op_namespacerN   Ú_higher_order_op_namespacerµ   r;   rR   r   r   r     s     ÿ ÿz_Ops.__init__c                 C   sJ   || j jkrt| j |ƒS |dkr&| jS t|ƒ}t| ||ƒ | j |¡ |S )NrG   )rÕ   rÐ   r¿   rÖ   rÊ   rÂ   rµ   r~   )r   r6   rP   r   r   r   rÄ   *  s    z_Ops.__getattr__c                 C   s
   t | jƒS r   rÅ   r;   r   r   r   rÇ   7  s    z_Ops.__iter__c              	   C   sB   t  ¡ rdS t |¡}tƒ � t |¡ W 5 Q R X | j |¡ dS )a  
        Loads a shared library from the given path into the current process.

        The library being loaded may run global initialization code to register
        custom operators with the PyTorch JIT runtime. This allows dynamically
        loading custom operators. For this, you should compile your operator
        and the static registration code into a shared library object, and then
        call ``torch.ops.load_library('path/to/libcustom.so')`` to load the
        shared object.

        After the library is loaded, it is added to the
        ``torch.ops.loaded_libraries`` attribute, a set that may be inspected
        for the paths of all libraries loaded using this function.

        Args:
            path (str): A path to a shared library to load.
        N)	r&   Z_running_with_deployr   Zresolve_library_pathr   r   ÚCDLLrÔ   r…   )r   Úpathr   r   r   Úload_library:  s    
z_Ops.load_library)	r<   r=   r>   rº   r   rÄ   rÇ   rÙ   rm   r   r   rR   r   rÒ     s
   rÒ   )4Ú
contextlibr   r+   r   ÚtypesÚtypingr   r   r   r   r   r   Ztorch._Cr&   r   Ztorch._functorch.pyfunctorchr	   r¬   r   Úcontextmanagerr   r   r'   Z_dispatch_is_included_in_aliasrA   r3   rD   rL   rN   rE   re   rf   rr   rs   r{   r”   r   r\   r‚   rƒ   rÓ   r„   r†   r‡   rˆ   r‰   r²   Ú
ModuleTyperÊ   rÏ   rÒ   rÑ   r   r   r   r   Ú<module>   sP     
c5q
		 PfB=