
    [Th                     V    S SK r S SKJr  S r " S S5      rS rg! \ a    Sr Nf = f)    N)_cudnnc                 f   U S:X  a#  [        [        R                  R                  5      $ U S:X  a#  [        [        R                  R                  5      $ U S:X  a#  [        [        R                  R
                  5      $ U S:X  a#  [        [        R                  R                  5      $ [        SU  35      e)NRNN_RELURNN_TANHLSTMGRUzUnknown mode: )intr   RNNModernn_relurnn_tanhlstmgru	Exception)modes    P/var/www/auris/envauris/lib/python3.13/site-packages/torch/backends/cudnn/rnn.pyget_cudnn_moder      s    z6>>**++		6>>**++	6>>&&''	6>>%%&&./00    c                   ,    \ rS rSrS rS rS rS rSrg)Unserializable   c                     Xl         g Ninner)selfr   s     r   __init__Unserializable.__init__   s    
r   c                     U R                   $ r   r   r   s    r   getUnserializable.get!   s    zzr   c                     g)Nz<unserializable> r   s    r   __getstate__Unserializable.__getstate__$   s     "r   c                     S U l         g r   r   )r   states     r   __setstate__Unserializable.__setstate__)   s	    
r   r   N)	__name__
__module____qualname____firstlineno__r   r    r$   r(   __static_attributes__r#   r   r   r   r      s    "
r   r   c                 v   S[        [        R                  R                  5       5      -   nU(       a  U OSnXC;  d  X4   R	                  5       cY  US:X  a  [        S 5      X4'   OE[        [        R                  " UUU[        R                  [        R                  " S5      S95      X4'   X4   R	                  5       nU$ )Ndesc_r   cuda)self_tydevice)	strtorchr1   current_devicer    r   _cudnn_init_dropout_stateuint8r3   )dropouttraindropout_seeddropout_statedropout_desc_name	dropout_p
dropout_tss          r   init_dropout_stater@   -   s    #ejj&?&?&A"BB aI.(,,.6>/=d/CM,/=// !KK <</0M, 1557Jr   )
torch.cudar5   torch._Cr   ImportErrorr   r   r@   r#   r   r   <module>rD      s=    
1   M   Fs    ((