+
    &j4                      h   ^ RI t  ! R R] P                  P                  4      t ! R R] P                  P                  4      t ! R R]4      t ! R R	]4      t ! R
 R]4      t ! R R] P                  P                  4      t ! R R] P                  P                  4      t	] P                  3R ltR# )    Nc                      a a ] tR t^t oV3R lV 3R llt]P                  P                  R 4       t]P                  P                  R 4       t	]P                  P                  R 4       t
RtVtV ;t# )MkldnnLinearc                   < V ^8  d   QhRR/#    returnN )format__classdict__s   "j/Users/jameslopez/projects/CWCArchive/cwc-podcast/.venv/lib/python3.14/site-packages/torch/utils/mkldnn.py__annotate__MkldnnLinear.__annotate__   s     [ [t [    c                  < \         SV `  4        V P                  R VP                  P	                  V4      4       VP
                  e-   V P                  RVP
                  P	                  4       4       R# V P                  R\        P                  ! VP                  P                  ^ 4      .\        P                  R7      P	                  4       4       R# )weightNbiasdtype)
super__init__register_bufferr   	to_mkldnnr   torchzerossizefloatselfdense_moduler   	__class__s   &&&r   r   MkldnnLinear.__init__   s    X|':':'D'DU'KL(   ):):)D)D)FG   \0055a89MWWY[r   c                    V P                   P                  4       V P                  P                  4       V P                  3# Nr   to_denser   trainingr   s   &r   __getstate__MkldnnLinear.__getstate__   -    $$&		(:(:(<dmmLLr   c                    V^ ,          P                  4       V n        V^,          P                  4       V n        V^,          V n        R# r   Nr   r   r   r&   r   states   &&r   __setstate__MkldnnLinear.__setstate__   5    Ah((*!H&&(	ar   c                   VP                   '       d   TMVP                  4       p\        P                  P                  P                  W P                  V P                  4      pVP                   '       d   TpV# VP                  4       pV# r#   )		is_mkldnnr   r   _C_nnmkldnn_linearr   r   r%   r   xx_mkldnny_mkldnnys   &&   r   forwardMkldnnLinear.forward   s`    188<<--hTYYOH *2):):)<r   r   r&   r   __name__
__module____qualname____firstlineno__r   r   jitscript_methodr(   r0   r=   __static_attributes____classdictcell____classcell__r    r   s   @@r   r   r      sr     [ [ YYM M YY! !
 YY  r   r   c                      a a ] tR t^%t oRt. ROtV3R lV 3R llt]P                  P                  R 4       t
]P                  P                  R 4       tRtVtV ;t# )_MkldnnConvNdz-Common base of MkldnnConv1d and MkldnnConv2d.c                   < V ^8  d   QhRR/# r   r	   )r
   r   s   "r   r   _MkldnnConvNd.__annotate__*   s     [ [ [r   c                  < \         SV `  4        VP                  V n        VP                  V n        VP                  V n        VP
                  V n        VP                  e-   V P                  RVP                  P                  4       4       R # V P                  R\        P                  ! VP                  P                  ^ 4      .\        P                  R7      P                  4       4       R # )Nr   r   )r   r   stridepaddingdilationgroupsr   r   r   r   r   r   r   r   r   r   r    s   &&r   r   _MkldnnConvNd.__init__*   s    "))#++$--"))(  ):):)D)D)FG
   \0055a89MWWY[r   c                    V P                   P                  4       V P                  P                  4       V P                  3# r#   r$   r'   s   &r   r(   _MkldnnConvNd.__getstate__<   r*   r   c           	         \         P                  ! VV P                  V P                  V P                  V P
                  V P                  V P                  4      # r#   )r   mkldnn_convolutionr   r   rQ   rP   rR   rS   r   r9   s   &&r   r=   _MkldnnConvNd.forward@   sB    ''KKIILLKKMMKK 	r   )rR   rS   rQ   rP   )rP   rQ   rR   rS   )rA   rB   rC   rD   __doc____constants__r   r   rE   rF   r(   r=   rG   rH   rI   rJ   s   @@r   rL   rL   %   s\     7?M[ [$ YYM M YY  r   rL   c                   v   a a ] tR t^Lt oV3R lV 3R llt]P                  P                  R 4       tRt	Vt
V ;t# )MkldnnConv1dc                   < V ^8  d   QhRR/# r   r	   )r
   r   s   "r   r   MkldnnConv1d.__annotate__M   s     M Mt Mr   c                |   < \         SV `  V4       V P                  R VP                  P	                  V4      4       R# r   Nr   r   r   r   r   r   s   &&&r   r   MkldnnConv1d.__init__M   s1    &X|':':'D'DU'KLr   c                    V^ ,          P                  4       V n        V^,          P                  4       V n        V^,          V n        R# r,   r-   r.   s   &&r   r0   MkldnnConv1d.__setstate__R   r2   r   r?   rA   rB   rC   rD   r   r   rE   rF   r0   rG   rH   rI   rJ   s   @@r   r_   r_   L   s4     M M
 YY! ! !r   r_   c                   v   a a ] tR t^Yt oV3R lV 3R llt]P                  P                  R 4       tRt	Vt
V ;t# )MkldnnConv2dc                   < V ^8  d   QhRR/# r   r	   )r
   r   s   "r   r   MkldnnConv2d.__annotate__Z         t r   c           
     "  < \         SV `  V4       V P                  R \        P                  P
                  P                  VP                  P                  V4      V P                  V P                  V P                  V P                  4      4       R# rc   )r   r   r   r   r5   r6   mkldnn_reorder_conv2d_weightr   r   rQ   rP   rR   rS   r   s   &&&r   r   MkldnnConv2d.__init__Z   e    &Xuxx||'P'P))%0LLKKMMKK( 	r   c                8   \         P                  P                  P                  V^ ,          P	                  4       V P
                  V P                  V P                  V P                  4      V n	        V^,          P	                  4       V n
        V^,          V n        R# r,   )r   r5   r6   ro   r   rQ   rP   rR   rS   r   r   r&   r.   s   &&r   r0   MkldnnConv2d.__setstate__d   i    hhll??!H LLKKMMKK !H&&(	ar   r?   rh   rJ   s   @@r   rj   rj   Y   2       YY! ! !r   rj   c                   v   a a ] tR t^ot oV3R lV 3R llt]P                  P                  R 4       tRt	Vt
V ;t# )MkldnnConv3dc                   < V ^8  d   QhRR/# r   r	   )r
   r   s   "r   r   MkldnnConv3d.__annotate__p   rm   r   c           
     "  < \         SV `  V4       V P                  R \        P                  P
                  P                  VP                  P                  V4      V P                  V P                  V P                  V P                  4      4       R# rc   )r   r   r   r   r5   r6   mkldnn_reorder_conv3d_weightr   r   rQ   rP   rR   rS   r   s   &&&r   r   MkldnnConv3d.__init__p   rq   r   c                8   \         P                  P                  P                  V^ ,          P	                  4       V P
                  V P                  V P                  V P                  4      V n	        V^,          P	                  4       V n
        V^,          V n        R# r,   )r   r5   r6   r{   r   rQ   rP   rR   rS   r   r   r&   r.   s   &&r   r0   MkldnnConv3d.__setstate__z   rt   r   r?   rh   rJ   s   @@r   rw   rw   o   ru   r   rw   c                      a a ] tR t^t oRR.tV3R lV 3R llt]P                  P                  R 4       t	]P                  P                  R 4       t
]P                  P                  R 4       tRtVtV ;t# )	MkldnnBatchNormexponential_average_factorepsc                   < V ^8  d   QhRR/# r   r	   )r
   r   s   "r   r   MkldnnBatchNorm.__annotate__   s     R R Rr   c                  < \         SV `  4        VP                  '       d   \        R 4      hVP                  '       g   \        R4      hVP
                  '       g   \        R4      hVP                  f	   RV n        MVP                  V n        VP                  V n        V P                  RVP                  P                  4       4       V P                  RVP                  P                  4       4       V P                  RVP                  P                  4       4       V P                  RVP                  P                  4       4       R# )	z4Only support eval mode batchnorm for mkldnn path nowz9Only support track_running_stats=True for mkldnn path nowz,Only support affine=True for mkldnn path nowNg        r   r   running_meanrunning_var)r   r   r&   AssertionErrortrack_running_statsaffinemomentumr   r   r   r   r   r   r   r   rT   s   &&r   r   MkldnnBatchNorm.__init__   s       !WXX/// !\]]""" !OPP  (.1D+.:.C.CD+##X|':':'D'D'FGV\%6%6%@%@%BC^\-F-F-P-P-RS]L,D,D,N,N,PQr   c                    V P                   P                  4       pV P                  P                  4       pV P                  P                  4       pV P                  P                  4       pWW4V P
                  3# r#   )r   r%   r   r   r   r&   )r   r   r   r   r   s   &    r   r(   MkldnnBatchNorm.__getstate__   s_    %%'yy!!#((113&&//1lGGr   c                   V^ ,          P                  4       V n        V^,          P                  4       V n        V^,          P                  4       V n        V^,          P                  4       V n        V^,          V n        R# r,   )r   r   r   r   r   r&   r.   s   &&r   r0   MkldnnBatchNorm.__setstate__   s_    Ah((*!H&&(	!!H..0 8--/ar   c                    \         P                  ! VV P                  V P                  V P                  V P
                  R V P                  V P                  R 4	      # )F)r   
batch_normr   r   r   r   r   r   rZ   s   &&r   r=   MkldnnBatchNorm.forward   sN    KKII++HH

 
	
r   )r   r   r   r   r   r&   r   )rA   rB   rC   rD   r]   r   r   rE   rF   r(   r0   r=   rG   rH   rI   rJ   s   @@r   r   r      s|     159MR R* YYH H YY! ! YY
 
 
r   r   c                      a a ] tR t^t oV3R lV 3R llt]P                  P                  R 4       t]P                  P                  R 4       t	]P                  P                  R 4       t
RtVtV ;t# )MkldnnPreluc                   < V ^8  d   QhRR/# r   r	   )r
   r   s   "r   r   MkldnnPrelu.__annotate__   s     M Mt Mr   c                z   < \         SV `  4        V P                  R VP                  P	                  V4      4       R# rc   rd   r   s   &&&r   r   MkldnnPrelu.__init__   s/    X|':':'D'DU'KLr   c                N    V P                   P                  4       V P                  3# r#   )r   r%   r&   r'   s   &r   r(   MkldnnPrelu.__getstate__   s    $$&66r   c                Z    V^ ,          P                  4       V n        V^,          V n        R# r,   )r   r   r&   r.   s   &&r   r0   MkldnnPrelu.__setstate__   s!    Ah((*ar   c                    VP                   '       d   TMVP                  4       p\        P                  ! W P                  4      pVP                   '       d   TpV# VP                  4       pV# r#   )r4   r   r   prelur   r%   r8   s   &&   r   r=   MkldnnPrelu.forward   sP    1;;x5H *2):):)<r   )r&   r   r@   rJ   s   @@r   r   r      sp     M M YY7 7 YY! ! YY  r   r   c                    aa V\         P                  \         P                  \         P                  39  d   \	        R 4      hR oVV3R loS! W4      # )z6MKLDNN only support float, bfloat16, and half path nowc                    \        V \        P                  P                  4      '       d   \	        W4      # \        V \        P                  P
                  4      '       d   \        W4      # \        V \        P                  P                  4      '       d   \        W4      # \        V \        P                  P                  4      '       d   \        W4      # \        V \        P                  P                  \        P                  P                  34      '       d   \        V 4      # \        V \        P                  P                  4      '       d   \        W4      # V # r#   )
isinstancer   nnLinearr   Conv1dr_   Conv2drj   Conv3drw   BatchNorm2dBatchNorm3dr   PReLUr   )mds   &&r   m_fnto_mkldnn.<locals>.m_fn   s    a))%%588??++%%588??++%%588??++%%EHH00%((2F2FGHH #1%%588>>**q$$Hr   c           	      n   < S! W4      pV P                  4        F  w  r4\        W#S! WA4      4       K  	  V# r#   )named_childrensetattr)r   r   new_mnamesub_mr   m_fn_recs   &&   r   r   to_mkldnn.<locals>.m_fn_rec   s5    Q
++-KDE%!34 .r   )r   r   bfloat16halfr   )moduler   r   r   s   &&@@r   r   r      sA    U[[%..%**==UVV$ F""r   )r   rE   ScriptModuler   rL   r_   rj   rw   r   r   r   r   r	   r   r   <module>r      s    599)) @$EII** $N
!= 
!!= !,!= !.4
eii,, 4
l%))(( * "KK #r   