a
    7Sic                     @   sj   U d dl mZ ddlmZ ddlZddlZg dZe Zee	 e
d< G dd dZdd
dZdddZdS )   )
OpOverload    )SetN)Libraryimpldefine_implsc                   @   s>   e Zd ZdZdddZdd ZdddZdd	d
Zdd ZdS )r   a  
    A class to create libraries that can be used to register new operators or
    override operators in existing libraries from Python.
    A user can optionally pass in a dispatch keyname if they only want to register
    kernels corresponding to only one specific dispatch key.

    Args:
        ns: library name
        kind: "DEF", "IMPL" (default: "IMPL")
        dispatch_key: PyTorch dispatch key (default: "")
     c                 C   sl   |dkr|dkrt d|tjddd }|j|j }}tj|||||| _|| _	t
 | _|| _|| _d S )NZIMPLZDEFzUnsupported kind:    )limitr   )
ValueError	tracebackextract_stackfilenamelinenotorch_C_dispatch_librarymnsset	_op_implskinddispatch_key)selfr   r   r   framer   r    r   I/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/library.py__init__   s    
zLibrary.__init__c                 C   s   d | j| j| jS )Nz)Library(kind={}, ns={}, dispatch_key={})>)formatr   r   r   )r   r   r   r   __repr__%   s    zLibrary.__repr__c                 C   s   |dkr| j }t|tr|}n8t|trN|jj}|jj}|dkrV|d | }ntd| jd |	dd  d | }|t
v rtd|	dd || j| j||| t
| | j| d S )Nr	   .zQimpl should be passed either a name or an OpOverload object as the first argument/z::zThis is not allowed since there's already a kernel registered from python overriding {}'s behavior for {} dispatch key and {} namespace.)r   
isinstancestrr   _schemanameoverload_nameRuntimeErrorr   splitr   r   r   r   addr   )r   op_namefnr   r'   r(   keyr   r   r   r   (   s$    

 
zLibrary.implc                 C   s   |dvrt d| j||S )aF  
        Takes a schema to define a new operator.
        Also, optionally takes `alias_analysis` argument to indicate if the aliasing properties of the arguments
        can be inferred from the schema (default behavior) or not ("CONSERVATIVE").

        Returns the name of the operator as inferred from the schema.
        )r	   ZFROM_SCHEMAZCONSERVATIVEzInvalid alias_analysis type)r)   r   r   )r   schemaalias_analysisr   r   r   r   B   s    
zLibrary.definec                 C   s   | j D ]}t| q| `d S N)r   r   remover   )r   r.   r   r   r   __del__P   s    
zLibrary.__del__N)r	   )r	   )r	   )	__name__
__module____qualname____doc__r   r    r   r   r3   r   r   r   r   r      s   


r   r	   c                    s    fdd}|S )Nc                    s    |   d S r1   )r   )fr   libr'   r   r   wrapX   s    zimpl.<locals>.wrapr   )r:   r'   r   r;   r   r9   r   r   W   s    r   c                    s    fdd}|S )Nc                    s     }||  d S r1   )r   r   )r8   r'   r0   r:   r/   r   r   r;   ]   s    zdefine.<locals>.wrapr   )r:   r/   r0   r;   r   r<   r   r   \   s    r   )r	   )r	   )_opsr   typingr   r   r   __all__r   r   r%   __annotations__r   r   r   r   r   r   r   <module>   s   I
