+
    "Ý³j‰  ã                   ób  a € R t0 t ^ RIHtHt ^ RIt^ RIt^ RIHt ]P                  P                  t
/ t] ^ k R t]! ]
P                  P                  ]
P                  P                  .4      RR l4       t]! ]
P"                  P                  .4      RR l4       t]! ]
P&                  P                  .4      RR l4       t]! ]
P*                  P                  .4      RR l4       t]! ]
P.                  P                  .4      RR	 l4       t]! ]
P2                  P                  .4      RR
 l4       tR# )é    )ÚAnyÚDictN)ÚSwizzleTensorc                ó   a € V 3R lpV# )z)Register aten ops to the swizzle op tablec                 ó,   <€ S F  pV \         V&   K  	  V # ©N)ÚSWIZZLE_OPS_TABLE)ÚfuncÚopÚaten_opss   & €Ú\/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchao/swizzle/swizzle_ops.pyÚ	decoratorÚimplements.<locals>.decorator   s   ø€ ÛˆBØ$(Ô˜bÓ!ñ àˆó    © )r   r   s   f r   Ú
implementsr      s   ø€ õð
 Ðr   c                 ó|  € V^ ,          pV^,          p\         P                  ! V4      '       dµ   \         P                  ! V4      '       d™   VP                  ^8X  dˆ   VP                  ^8X  dw   RpRp\        V\        4      '       d   VP                  4       pRp\        V\        4      '       d   VP                  4       pRp\        P                  P                  W4WV4      pV# \        V\        4      '       d   VP                  4       MTp\        V\        4      '       d   VP                  4       MTpV ! W43/ VB pV# )r   FT)
ÚtorchÚis_floating_pointÚndimÚ
isinstancer   Ú	as_tensorÚtorchaoÚopsÚ
swizzle_mmÚ	unswizzle)Úaten_opÚargsÚkwargsÚaÚbÚa_is_swizzledÚb_is_swizzledÚ
tensor_outs   &&&     r   r   r      sù   € àˆQ�€AØˆQ�€Aô 	×Ò ×"Ò"Ü×#Ò# A×&Ò&Ø�F‰F�aŒKØ�F‰F�aŒKàˆØˆÜ�aœ×'Ò'Ø—‘“ˆAØ ˆMÜ�aœ×'Ò'Ø—‘“ˆAØ ˆMÜ—[‘[×+Ñ+¨A°-ÓOˆ
ð
 Ðô (¨¬=×9Ò9ˆA�K‰KŒM¸qˆÜ'¨¬=×9Ò9ˆA�K‰KŒM¸qˆÙ˜QÑ, VÑ,ˆ
ØÐr   c                 óØ   € V^ ,          pV^,          p\        V\        4      '       d   VP                  4       MTp\        V\        4      '       d   VP                  4       MTpV ! W43/ VB # ©r   ©r   r   r   )r   r   r   r    r!   s   &&&  r   Úswizzle_bmmr(   7   sV   € àˆQ�€AØˆQ�€Aä# A¤}×5Ò5ˆ�‰Œ¸1€AÜ# A¤}×5Ò5ˆ�‰Œ¸1€AÙ�1Ñ"˜6Ñ"Ð"r   c                 óú   € V^ ,          pV^,          pV^,          p\        V\        4      '       d   VP                  4       MTp\        V\        4      '       d   VP                  4       MTpV ! W4WQR,          3/ VB # )r   :é   NNr'   )r   r   r   Úbiasr    r!   s   &&&   r   Úswizzle_addmmr,   A   se   € à��7€DØˆQ�€AØˆQ�€AÜ# A¤}×5Ò5ˆ�‰Œ¸1€AÜ# A¤}×5Ò5ˆ�‰Œ¸1€AÙ�4˜A B�xÑ2¨6Ñ2Ð2r   c                 óæ  € V^ ,          pV^,          pV^,          pV^,          p\        V4      ^8:  d   RMV^,          p\        V4      ^8:  d   RMV^,          p\        V4      ^8:  d   RMV^,          p	Rp
Rp\        V\        4      '       d   VP                  4       pRp
\        V\        4      '       d   VP                  4       pRp\        P
                  P                  ! VVV
VVVVVV	3	/ VB # )r   NFT)Úlenr   r   r   r   r   Úswizzle_scaled_mm)r   r   r   r    r!   Úscale_aÚscale_br+   Úscale_resultÚ	out_dtyper"   r#   s   &&&         r   r/   r/   K   så   € àˆQ�€AØˆQ�€AØ�1�g€GØ�1�g€GÜ�t“9 ”>‰4 t¨A¥w€DÜ˜t›9¨œ>‘4¨t°A­w€LÜ˜D›	 Qœ‘¨D°­G€Ià€MØ€MÜ�!”]×#Ò#Ø�K‰K‹MˆØˆÜ�!”]×#Ò#Ø�K‰K‹MˆØˆÜ�;‰;×(Ò(Ø	Ø	ØØØØØØØñð ñð r   c                 óÌ   € V^ ,          pV^,          p\        V4      ^8X  d-   V^ ,          ^8X  d   V^,          ^ 8X  d   VP                  4       # V ! VP                  4       V4      # r&   )r.   Úshallow_transposer   )r   r   r   ÚtensorÚdimss   &&&  r   Úswizzle_permuter8   k   sX   € à�!�W€FØ��7€DÜ
ˆ4ƒy�A„~˜$˜q�' Qœ,¨4°­7°a¬<Ø×'Ñ'Ó)Ð)Ù�6×#Ñ#Ó% tÓ,Ð,r   c                 ó4   € V^ ,          pVP                  4       # r&   )r5   )r   r   r   r6   s   &&& r   Úswizzle_numpy_Tr:   t   s   € à�!�W€FØ×#Ñ#Ó%Ð%r   c                ób   € V ^8„  d   Qh/ ^ \         9   d   \        \        \        3,          ;R&   # )é   r	   )Ú__conditional_annotations__r   r   )Úformats   "r   Ú__annotate__r?      s"   € × Ñ ÷ 'Ò &”4œœS˜•>Ñ &ò r   r   )r=   Útypingr   r   r   Útorchao.opsr   Útorchao.swizzle.swizzle_tensorr   r   Úatenr	   r   ÚmmÚdefaultÚmatmulr   Úbmmr(   Úaddmmr,   Ú
_scaled_mmr/   Úpermuter8   Únumpy_Tr:   r?   )r=   s   @r   Ú<module>rL      s   øð÷ Ô ã ã Ý 8à‡y�y‡~�~€Ø$&Ð Ó &òñ ˆT�W‰W�_‰_˜dŸk™k×1Ñ1Ð2Ó3óó 4ðñ4 ˆT�X‰X×ÑÐÓó#ó  ð#ñ ˆT�Z‰Z×ÑÐ Ó!ó3ó "ð3ñ ˆT�_‰_×$Ñ$Ð%Ó&óó 'ðñ> ˆT�\‰\×!Ñ!Ð"Ó#ó-ó $ð-ñ ˆT�\‰\×!Ñ!Ð"Ó#ó&ó $ò&r   