
    sg                     8    U d dl Z daeed<   defdZdeddfdZy)    NT_is_fastpath_enabledreturnc                  L    t         j                  j                         st        S y)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)torchjitis_scriptingr        N/var/www/html/venv/lib/python3.12/site-packages/torch/backends/mha/__init__.pyget_fastpath_enabledr   	   s     99!!###r
   valuec                     | a y)z!Sets whether fast path is enabledN)r   )r   s    r   set_fastpath_enabledr      s
     !r
   )r   r   bool__annotations__r   r   r	   r
   r   <module>r      s6     " d !
d 
! ! !r
   