+
    ÒÜ³jÓ  ã                   óN   € ^ RI t . R
Ot. ROtR R ltR R ltR R ltR tR	 tR# )é    Nc                ód   € V ^8„  d   QhR\         P                  R\         P                  R\        /# )é   ÚqweightÚqzerosÚbits©ÚtorchÚTensorÚint)Úformats   "ÚX/home/luca/screen-checking/.venv/lib/python3.14/site-packages/awq/utils/packing_utils.pyÚ__annotate__r      s)   € ÷ ñ œŸ™ð ¬e¯l©lð Ä#ñ ó    c                 óô  € \         P                  ! ^ ^ W!P                  R7      p\         P                  ! V R,          VR,          4      P	                  \         P
                  4      pVP                  VP                  ^ ,          R4      pVei   \         P                  ! VR,          VR,          4      P	                  \         P
                  4      pVP                  VP                  ^ ,          R4      pWE3# TpWE3# )r   ©Údevice)ºNNNr   N©NNr   éÿÿÿÿ)r	   Úaranger   Úbitwise_right_shiftÚtoÚint8ÚviewÚshape)r   r   r   ÚshiftsÚiweightsÚizeross   &&&   r   Ú
unpack_awqr      sÎ   € Ü�\Š\˜!˜R ¯m©mÔ<€Fô ×(Ò(¨°Õ)<¸fÀ]Õ>SÓT×WÑWÜ�
‰
ó€Hð �}‰}˜XŸ^™^¨AÕ.°Ó3€Hð ÒÜ×*Ò*¨6°*Õ+=¸vÀmÕ?TÓU×XÑXÜ�J‰Jó
ˆð —‘˜VŸ\™\¨!�_¨bÓ1ˆð ÐÐð ˆàÐÐr   c                ód   € V ^8„  d   QhR\         P                  R\         P                  R\        /# ©r   r   r   r   r   )r   s   "r   r   r      s)   € ÷ ñ ¤§¡ð ´e·l±lð Ì#ñ r   c                 ó2  € \         P                  ! V P                  R,          \         P                  VP                  R7      pVP                  R^ V,          4      pVR\        3,          pVP                  R4      pVe   VRV3,          pV RV3,          p W3# )é   )Údtyper   r   r   )r	   r   r   Úint32r   r   ÚAWQ_REVERSE_ORDER)r   r   r   Úreverse_order_tensors   &&& r   Úreverse_awq_orderr(      s•   € Ü Ÿ<š<Ø�‰�rÕÜ�k‰kØ�}‰}ôÐð
 0×4Ñ4°R¸¸t½ÓDÐØ/°Ô3DÐ0DÕEÐØ/×4Ñ4°RÓ8ÐàÒØ˜Ð/Ð/Õ0ˆØ˜Ð/Ð/Õ0€HàÐÐr   c                ód   € V ^8„  d   QhR\         P                  R\         P                  R\        /# r!   r   )r   s   "r   r   r   .   s)   € ÷ ñ œ5Ÿ<™<ð ´·±ð ÄSñ r   c                 óZ  € \         P                  ! ^ ^ W P                  R7      pV P                  V P                  ^ ,          ^ V,          ,          ^ V,          R4      p \         P
                  ! WR,          4      P                  ^R7      P                  \         P                  4      pVP                  RVP                  ^,          ^ V,          ,          ^ V,          4      p\         P
                  ! WR,          4      P                  RR7      P                  \         P                  4      pWE3# )r   r   ©Údimr   )Nr   Nr   )	r	   r   r   r   r   Úbitwise_left_shiftÚsumr   r%   )r   r   r   r   r   r   s   &&&   r   Úpack_exllamar/   .   sÕ   € Ü�\Š\˜!˜R ¯o©oÔ>€Fð �}‰}˜XŸ^™^¨AÕ.°2¸µ:Õ>ÀÀdÅ
ÈBÓO€Hä× Ò  °-Õ+@ÓAß	‰�ˆ‹ß	‰ŒE�K‰K‹ð ð �[‰[˜˜VŸ\™\¨!�_°°tµÕ<¸bÀD½jÓI€Fä× Ò  °Õ)>Ó?ß	‰�ˆ‹ß	‰ŒE�K‰K‹ð ð ˆ?Ðr   c                 ó  € \        WV4      w  r4\        W4V4      w  r4\        P                  ! V^V,          ^,
          4      p\        P                  ! V^V,          ^,
          4      pV^,
          p\	        W4V4      w  rW3# )r   )r   r(   r	   Úbitwise_andr/   )r   r   r   Úiweightr   s   &&&  r   Úunpack_reorder_packr3   D   su   € ä  °$Ó7�O€Gä'¨¸Ó>�O€Gô ×Ò ¨!¨T­'°Q­Ó7€GÜ×Ò˜v¨¨4­°1¥}Ó5€Fð �a�Z€Fä" 7°DÓ9�O€Gàˆ?Ðr   c                 ó<  € \        WV4      w  rV\        WVV4      w  rV\        P                  ! V^V,          ^,
          4      p\        P                  ! V^V,          ^,
          4      pVP	                  V^ R7      pVP	                  V^ R7      pWV,
          V,          pV# )r   r+   )r   r(   r	   r1   Úrepeat_interleave)r   r   Úscalesr   Ú
group_sizer2   r   s   &&&&&  r   Údequantize_gemmr8   W   s’   € ä  °$Ó7�O€Gä'¨¸Ó>�O€Gô ×Ò ¨!¨T­'°Q­Ó7€GÜ×Ò˜v¨¨4­°1¥}Ó5€Fð ×%Ñ% j°aÐ%Ó8€FØ×%Ñ% j°aÐ%Ó8€FØÕ 6Õ)€Gà€Nr   )r   r   é   é   r#   é   é   é   )r   r9   r#   r<   r   r:   r;   r=   )r	   Ú	AWQ_ORDERr&   r   r(   r/   r3   r8   © r   r   Ú<module>r@      s.   ðÛ ò %€	Ú,Ð õõ*õ"ò,ô&r   