+
    ˜°³jY'  ã                   ól   € ^ RI t ^ RIt] P                  ! R4      R 4       tR t]P
                  3R ltR# )é    Nc                  ó|  €  ! R  R\         P                  P                  4      p  ! R R\         P                  P                  4      p ! R RV4      p ! R RV4      p ! R R	V4      p ! R
 R\         P                  P                  4      p ! R R\         P                  P                  4      pRV RVRVRVR	VRVRV/# )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# )Ú)_get_mkldnn_classes.<locals>.MkldnnLinearc                ó   <€ V ^8„  d   QhRR/# ©é   ÚreturnN© )ÚformatÚ__classdict__s   "€ÚS/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/utils/mkldnn.pyÚ__annotate__Ú6_get_mkldnn_classes.<locals>.MkldnnLinear.__annotate__   s   ø€ ÷ 	_ñ 	_°4ñ 	_ó    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   Ú2_get_mkldnn_classes.<locals>.MkldnnLinear.__init__   sŸ   ø€ Ü‰GÑÔØ× Ñ  ¨<×+>Ñ+>×+HÑ+HÈÓ+OÔPØ× Ñ Ò,ð ×$Ñ$ V¨\×->Ñ->×-HÑ-HÓ-JÖKð ×$Ñ$ØÜ—K’K ×!4Ñ!4×!9Ñ!9¸!Ó!<Ð =ÄUÇ[Á[ÔQ×[Ñ[Ó]ö_r   c                ó€   € V P                   P                  4       V P                  P                  4       V P                  3# ©N©r   Úto_denser   Útraining©r   s   &r   Ú__getstate__Ú6_get_mkldnn_classes.<locals>.MkldnnLinear.__getstate__   ó-   € à—K‘K×(Ñ(Ó*¨D¯I©I×,>Ñ,>Ó,@À$Ç-Á-ÐPÐPr   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__Ú6_get_mkldnn_classes.<locals>.MkldnnLinear.__setstate__   ó5   € à �(×,Ñ,Ó.ˆDŒKØ˜a�×*Ñ*Ó,ˆDŒIØ! !�HˆDŽMr   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Ú1_get_mkldnn_classes.<locals>.MkldnnLinear.forward%   s`   € àŸKŸK˜K‘q¨Q¯[©[«]ˆHÜ—x‘x—|‘|×1Ñ1°(¿K¹KÈÏÉÓSˆHØŸKŸK˜K�ˆAØˆHð .6×->Ñ->Ó-@ˆAØˆHr   ©r   r'   r   ©Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r   ÚjitÚscript_methodr)   r1   r>   Ú__static_attributes__Ú__classdictcell__Ú__classcell__©r!   r   s   @@r   ÚMkldnnLinearr      sr   ù‡ € ÷	_ó 	_ð 
�‰×	 Ñ	 ñ	Qó 
!ð	Qð 
�‰×	 Ñ	 ñ	%ó 
!ð	%ð
 
�‰×	 Ñ	 ñ	ó 
!÷	ð 	r   rL   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# )Ú*_get_mkldnn_classes.<locals>._MkldnnConvNdz-Common base of MkldnnConv1d and MkldnnConv2d.c                ó   <€ V ^8„  d   QhRR/# r   r
   )r   r   s   "€r   r   Ú7_get_mkldnn_classes.<locals>._MkldnnConvNd.__annotate__1   s   ø€ ÷ 	_ñ 	_¨Dñ 	_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   Ú3_get_mkldnn_classes.<locals>._MkldnnConvNd.__init__1   s°   ø€ Ü‰GÑÔà&×-Ñ-ˆDŒKØ'×/Ñ/ˆDŒLØ(×1Ñ1ˆDŒMØ&×-Ñ-ˆDŒKà× Ñ Ò,Ø×$Ñ$ V¨\×->Ñ->×-HÑ-HÓ-JÖKð
 ×$Ñ$ØÜ—K’K ×!4Ñ!4×!9Ñ!9¸!Ó!<Ð =ÄUÇ[Á[ÔQ×[Ñ[Ó]ö_r   c                ó€   € V P                   P                  4       V P                  P                  4       V P                  3# r$   r%   r(   s   &r   r)   Ú7_get_mkldnn_classes.<locals>._MkldnnConvNd.__getstate__C   r+   r   c           	     ó²   € \         P                  ! VV P                  V P                  V P                  V P
                  V P                  V P                  4      # r$   )r   Úmkldnn_convolutionr   r   rS   rR   rT   rU   ©r   r:   s   &&r   r>   Ú2_get_mkldnn_classes.<locals>._MkldnnConvNd.forwardG   sB   € ä×+Ò+ØØ—‘Ø—	‘	Ø—‘Ø—‘Ø—‘Ø—‘óð r   )rT   rU   rS   rR   )rR   rS   rT   rU   )rB   rC   rD   rE   Ú__doc__Ú__constants__r   r   rF   rG   r)   r>   rH   rI   rJ   rK   s   @@r   Ú_MkldnnConvNdrN   ,   s\   ù‡ € Ù;âCˆ÷	_ó 	_ð$ 
�‰×	 Ñ	 ñ	Qó 
!ð	Qð 
�‰×	 Ñ	 ñ	ó 
!÷	ð 	r   r`   c                   óv   a a€ ] tR t^Rt oV3R lV 3R llt]P                  P                  R 4       tRt	Vt
V ;t# )Ú)_get_mkldnn_classes.<locals>.MkldnnConv1dc                ó   <€ V ^8„  d   QhRR/# r   r
   )r   r   s   "€r   r   Ú6_get_mkldnn_classes.<locals>.MkldnnConv1d.__annotate__S   s   ø€ ÷ 	Qñ 	Q°4ñ 	Q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   Ú2_get_mkldnn_classes.<locals>.MkldnnConv1d.__init__S   s1   ø€ Ü‰GÑ˜\Ô*à× Ñ  ¨<×+>Ñ+>×+HÑ+HÈÓ+OÖPr   c                ó’   € V^ ,          P                  4       V n        V^,          P                  4       V n        V^,          V n        R# r-   r.   r/   s   &&r   r1   Ú6_get_mkldnn_classes.<locals>.MkldnnConv1d.__setstate__X   r3   r   r@   ©rB   rC   rD   rE   r   r   rF   rG   r1   rH   rI   rJ   rK   s   @@r   ÚMkldnnConv1drb   R   s4   ù‡ € ÷	Qó 	Qð
 
�‰×	 Ñ	 ñ	%ó 
!÷	%ð 	%r   rl   c                   óv   a a€ ] tR t^^t oV3R lV 3R llt]P                  P                  R 4       tRt	Vt
V ;t# )Ú)_get_mkldnn_classes.<locals>.MkldnnConv2dc                ó   <€ V ^8„  d   QhRR/# r   r
   )r   r   s   "€r   r   Ú6_get_mkldnn_classes.<locals>.MkldnnConv2d.__annotate___   ó   ø€ ÷ 	ñ 	°4ñ 	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# rf   )r   r   r   r   r6   r7   Úmkldnn_reorder_conv2d_weightr   r   rS   rR   rT   rU   r   s   &&&€r   r   Ú2_get_mkldnn_classes.<locals>.MkldnnConv2d.__init___   óe   ø€ Ü‰GÑ˜\Ô*à× Ñ  ¬5¯8©8¯<©<×+TÑ+TØ×#Ñ#×-Ñ-¨eÓ4Ø—‘Ø—‘Ø—‘Ø—‘ó,ö 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   r6   r7   rs   r   rS   rR   rT   rU   r   r   r'   r/   s   &&r   r1   Ú6_get_mkldnn_classes.<locals>.MkldnnConv2d.__setstate__i   ói   € äŸ(™(Ÿ,™,×CÑCØ�a•×"Ñ"Ó$Ø—‘Ø—‘Ø—‘Ø—‘óˆDŒKð ˜a�×*Ñ*Ó,ˆDŒIØ! !�HˆDŽMr   r@   rk   rK   s   @@r   ÚMkldnnConv2drn   ^   ó2   ù‡ € ÷	ó 	ð 
�‰×	 Ñ	 ñ	%ó 
!÷	%ð 	%r   ry   c                   óv   a a€ ] tR t^tt oV3R lV 3R llt]P                  P                  R 4       tRt	Vt
V ;t# )Ú)_get_mkldnn_classes.<locals>.MkldnnConv3dc                ó   <€ V ^8„  d   QhRR/# r   r
   )r   r   s   "€r   r   Ú6_get_mkldnn_classes.<locals>.MkldnnConv3d.__annotate__u   rq   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# rf   )r   r   r   r   r6   r7   Úmkldnn_reorder_conv3d_weightr   r   rS   rR   rT   rU   r   s   &&&€r   r   Ú2_get_mkldnn_classes.<locals>.MkldnnConv3d.__init__u   ru   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   r6   r7   r€   r   rS   rR   rT   rU   r   r   r'   r/   s   &&r   r1   Ú6_get_mkldnn_classes.<locals>.MkldnnConv3d.__setstate__   rx   r   r@   rk   rK   s   @@r   ÚMkldnnConv3dr|   t   rz   r   r„   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# )	Ú,_get_mkldnn_classes.<locals>.MkldnnBatchNormÚexponential_average_factorÚepsc                ó   <€ V ^8„  d   QhRR/# r   r
   )r   r   s   "€r   r   Ú9_get_mkldnn_classes.<locals>.MkldnnBatchNorm.__annotate__�   s   ø€ ÷ 	Vñ 	V¨Dñ 	V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�   rV   s   &&€r   r   Ú5_get_mkldnn_classes.<locals>.MkldnnBatchNorm.__init__�   s  ø€ Ü‰GÑÔà×$×$Ð$Ü$Ð%[Ó\Ð\Ø×3×3Ð3Ü$Ð%`ÓaÐaØ×&×&Ð&Ü$Ð%SÓTÐTà×$Ñ$Ò,Ø25�Õ/à2>×2GÑ2G�Ô/Ø#×'Ñ'ˆDŒHà× Ñ  ¨<×+>Ñ+>×+HÑ+HÓ+JÔKØ× Ñ  ¨×):Ñ):×)DÑ)DÓ)FÔGØ× Ñ  °×1JÑ1J×1TÑ1TÓ1VÔWØ× Ñ  °×0HÑ0H×0RÑ0RÓ0TÖUr   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)   Ú9_get_mkldnn_classes.<locals>.MkldnnBatchNorm.__getstate__¢   s_   € à—[‘[×)Ñ)Ó+ˆFØ—9‘9×%Ñ%Ó'ˆDØ×,Ñ,×5Ñ5Ó7ˆLØ×*Ñ*×3Ñ3Ó5ˆKØ ,¸T¿]¹]ÐKÐKr   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   r1   Ú9_get_mkldnn_classes.<locals>.MkldnnBatchNorm.__setstate__ª   s_   € à �(×,Ñ,Ó.ˆDŒKØ˜a�×*Ñ*Ó,ˆDŒIØ % a¥× 2Ñ 2Ó 4ˆDÔØ$ Q�x×1Ñ1Ó3ˆDÔØ! !�HˆDŽMr   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ˆ   r\   s   &&r   r>   Ú4_get_mkldnn_classes.<locals>.MkldnnBatchNorm.forward²   sN   € ä×#Ò#ØØ—‘Ø—	‘	Ø×!Ñ!Ø× Ñ ØØ×/Ñ/Ø—‘Øó
ð 
r   )r   rˆ   r‡   rŒ   r�   r'   r   )rB   rC   rD   rE   r_   r   r   rF   rG   r)   r1   r>   rH   rI   rJ   rK   s   @@r   ÚMkldnnBatchNormr†   Š   s|   ù‡ € Ø5°uÐ=ˆ÷	Vó 	Vð* 
�‰×	 Ñ	 ñ	Ló 
!ð	Lð 
�‰×	 Ñ	 ñ	%ó 
!ð	%ð 
�‰×	 Ñ	 ñ	ó 
!÷	ð 	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# )Ú(_get_mkldnn_classes.<locals>.MkldnnPreluc                ó   <€ V ^8„  d   QhRR/# r   r
   )r   r   s   "€r   r   Ú5_get_mkldnn_classes.<locals>.MkldnnPrelu.__annotate__Á   s   ø€ ÷ 	Qñ 	Q°4ñ 	Qr   c                óz   <€ \         SV `  4        V P                  R VP                  P	                  V4      4       R# rf   rg   r   s   &&&€r   r   Ú1_get_mkldnn_classes.<locals>.MkldnnPrelu.__init__Á   s/   ø€ Ü‰GÑÔØ× Ñ  ¨<×+>Ñ+>×+HÑ+HÈÓ+OÖPr   c                óN   € V P                   P                  4       V P                  3# r$   )r   r&   r'   r(   s   &r   r)   Ú5_get_mkldnn_classes.<locals>.MkldnnPrelu.__getstate__Å   s   € à—K‘K×(Ñ(Ó*¨D¯M©MÐ:Ð:r   c                óZ   € V^ ,          P                  4       V n        V^,          V n        R# r-   )r   r   r'   r/   s   &&r   r1   Ú5_get_mkldnn_classes.<locals>.MkldnnPrelu.__setstate__É   s!   € à �(×,Ñ,Ó.ˆDŒKØ! !�HˆDŽMr   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$   )r5   r   r   Úprelur   r&   r9   s   &&   r   r>   Ú0_get_mkldnn_classes.<locals>.MkldnnPrelu.forwardÎ   sP   € àŸKŸK˜K‘q¨Q¯[©[«]ˆHÜ—{’{ 8¯[©[Ó9ˆHØŸKŸK˜K�ˆAØˆHð .6×->Ñ->Ó-@ˆAØˆHr   )r'   r   rA   rK   s   @@r   ÚMkldnnPrelurœ   À   sp   ù‡ € ÷	Qó 	Qð 
�‰×	 Ñ	 ñ	;ó 
!ð	;ð 
�‰×	 Ñ	 ñ	%ó 
!ð	%ð 
�‰×	 Ñ	 ñ	ó 
!÷	ð 	r   r¨   )r   rF   ÚScriptModule)rL   r`   rl   ry   r„   rš   r¨   s          r   Ú_get_mkldnn_classesrª      s¯   € ô”u—y‘y×-Ñ-ô ô>$œŸ	™	×.Ñ.ô $ôL
%�}ô 
%ô%�}ô %ô,%�}ô %ô,4œ%Ÿ)™)×0Ñ0ô 4ôl”e—i‘i×,Ñ,ô ð, 	˜Ø˜Ø˜Ø˜Ø˜Ø˜?Ø�{ðð r   c                 ó`   € \        4       pW9   d	   W,          # \        R \        : RV : 24      h)zmodule z has no attribute )rª   ÚAttributeErrorrB   )ÚnameÚclassess   & r   Ú__getattr__r¯   à   s0   € Ü!Ó#€GØ„Ø�}ÐÜ
˜7¤8¡,Ð.@ÀÁÐIÓ
JÐJr   c                 óD  aaaaaaa	a
€ V\         P                  \         P                  \         P                  39  d   \	        R 4      h\        4       pVR,          oVR,          oVR,          oVR,          oVR,          oVR,          oVVVVVV3R lo	V	V
3R lo
S
! W4      # )	z6MKLDNN only support float, bfloat16, and half path nowrL   rl   ry   r„   rš   r¨   c                 ó”  <€ \        V \        P                  P                  4      '       d	   S! W4      # \        V \        P                  P                  4      '       d	   S! W4      # \        V \        P                  P
                  4      '       d	   S! W4      # \        V \        P                  P                  4      '       d	   S! W4      # \        V \        P                  P                  \        P                  P                  34      '       d	   S! V 4      # \        V \        P                  P                  4      '       d	   S! W4      # V # r$   )
Ú
isinstancer   ÚnnÚLinearÚConv1dÚConv2dÚConv3dÚBatchNorm2dÚBatchNorm3dÚPReLU)ÚmÚdrš   rl   ry   r„   rL   r¨   s   &&€€€€€€r   Úm_fnÚto_mkldnn.<locals>.m_fnó   sÛ   ø€ Ü�aœŸ™Ÿ™×)Ò)Ù Ó%Ð%Ü˜œ5Ÿ8™8Ÿ?™?×+Ò+Ù Ó%Ð%Ü˜œ5Ÿ8™8Ÿ?™?×+Ò+Ù Ó%Ð%Ü˜œ5Ÿ8™8Ÿ?™?×+Ò+Ù Ó%Ð%Ü˜œEŸH™H×0Ñ0´%·(±(×2FÑ2FÐG×HÒHñ # 1Ó%Ð%Ü˜œ5Ÿ8™8Ÿ>™>×*Ò*Ù˜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_mr­   Úsub_mr½   Úm_fn_recs   &&   €€r   rÄ   Úto_mkldnn.<locals>.m_fn_rec  s5   ø€ Ù�Q“
ˆØ×+Ñ+Ö-‰KˆDÜ�E¡¨%Ó!3Ö4ñ .àˆr   )r   r   Úbfloat16ÚhalfrŽ   rª   )Úmoduler   r®   rš   rl   ry   r„   rL   r¨   r½   rÄ   s   && @@@@@@@@r   r   r   ç   s‹   ÿ€ Ø”U—[‘[¤%§.¡.´%·*±*Ð=Ô=ÜÐUÓVÐVä!Ó#€GØ˜>Õ*€LØ˜>Õ*€LØ˜>Õ*€LØ˜>Õ*€LØÐ/Õ0€OØ˜-Õ(€K÷ò ö$ñ �FÓ"Ð"r   )Ú	functoolsr   Ú	lru_cacherª   r¯   r   r   r
   r   r   Ú<module>rË      sA   ðã ã ð ×Ò�TÓñQó ðQòhKð "ŸK™Kö $#r   