U
    yh                     @   s8   U d dl Z daeed< edddZedddd	ZdS )
    NT_is_fastpath_enabled)returnc                   C   s   t j stS dS )a  Returns whether fast path for TransformerEncoder and MultiHeadAttention
    is enabled, or ``True`` if jit is scripting.

    ..note:
        The fastpath might not be run even if ``get_fastpath_enabled`` returns
        ``True`` unless all conditions on inputs are met.
    T)torchZjitZis_scriptingr    r   r   M/var/www/html/venv/lib/python3.8/site-packages/torch/backends/mha/__init__.pyget_fastpath_enabled   s    
r   )valuer   c                 C   s   | a dS )z!Sets whether fast path is enabledN)r   )r   r   r   r   set_fastpath_enabled   s    r	   )r   r   bool__annotations__r   r	   r   r   r   r   <module>   s    