+
    "Ý³j¯  ã                  ó*  a € 0 t $ R t^ RIHt ^ RIt^ RIt^ RIHt ^ RIH	t	 ^ RI
Ht ^ RIHt ^ RIt^RIHt ]! 4       t]	! RR	R
7      t]P&                  ! 4       t. tR]R&   R R lt]R R l4       tR tR tR R ltR R ltR R ltR R lt. ROtR# )zBStructured device telemetry scoped to one quantization invocation.)ÚannotationsN)Úcontextmanager)Ú
ContextVar©Úwraps)ÚAny)Úsetup_loggerÚgptqmodel_device_telemetryF)Údefaultúlist[dict[str, Any]]Ú_recordsc               ó   € V ^8„  d   QhRR/# )é   ÚreturnÚbool© )Úformats   "Úa/home/luca/screen-checking/.venv/lib/python3.14/site-packages/gptqmodel/utils/device_telemetry.pyÚ__annotate__r      s   € ÷ ñ  $ñ ó    c                 ó*   € \         P                  4       # )z8Return ``True`` when device telemetry should be emitted.)Ú_enabledÚgetr   r   r   Údevice_telemetry_enabledr      s   € ô �<‰<‹>Ðr   c               ó   € V ^8„  d   QhRR/# )r   Úenabledr   r   )r   s   "r   r   r   #   s   € ÷ ñ  Dñ r   c              #  óÜ   "  € \        V 4      \        Jd   \        R 4      h\        P	                  V 4      p Rx € \        P                  V4       R#   \        P                  T4       i ; i5i)z"device_telemetry must be a booleanN)Útyper   Ú
ValueErrorr   ÚsetÚreset)r   Útokens   & r   Údevice_telemetry_scoper"   "   sK   é € äˆGƒ}œDÓ ÜÐ=Ó>Ð>Ü�L‰L˜Ó!€EðÛä�‰�uÖøŒ�‰�uÕüs   ‚4A,·A »A,ÁA)Á)A,c                ó0   a € \        S 4      V 3R  l4       pV# )c                óÊ   <€ \        \        V R R4      RR4      p\        VRR4      p\        V4      ;_uu_ 4        S! V .VO5/ VB uuRRR4       #   + '       g   i     R# ; i)Úquantize_configNÚ	telemetryÚdeviceF)Úgetattrr"   )ÚmodelÚargsÚkwargsr&   r   Úfns   &*,  €r   ÚrunÚ/with_quantization_device_telemetry.<locals>.run.   sV   ø€ äœG EÐ+<¸dÓCÀ[ÐRVÓWˆ	Ü˜) X¨uÓ5ˆÜ# G×,Õ,Ù�eÐ-˜dÒ- fÑ-÷ -×,×,Ó,ús   ºAÁA"	r   )r,   r-   s   f r   Ú"with_quantization_device_telemetryr/   -   s    ø€ Ü
ˆ2ƒYô.ó ð.ð €Jr   c                óH   a a€ \        4       o\        S 4      VV 3R l4       pV# )zKPropagate only this setting to queued work; reset reused workers afterward.c                 óx   <€ \        S4      ;_uu_ 4        S! V / VB uuR R R 4       #   + '       g   i     R # ; i)N)r"   )r*   r+   r   r,   s   *,€€r   r-   Ú%capture_device_telemetry.<locals>.run<   s*   ø€ ä# G×,Õ,Ù�tÐ&˜vÑ&÷ -×,×,Ó,ús   •(¨9	)r   r   )r,   r-   r   s   f @r   Úcapture_device_telemetryr3   8   s(   ù€ ä&Ó(€Gä
ˆ2ƒYõ'ó ð'ð €Jr   c               ó    € V ^8„  d   QhRRRR/# )r   Úvaluer   r   r   )r   s   "r   r   r   D   s   € ÷ ñ ˜Cð  Cñ r   c                óä  € \        V \        P                  4      '       d   \        V 4      # \        V \        P                  4      '       d   \        V P                  4      # \        V \
        \        34      '       d   V  Uu. uF  p\        V4      NK  	  up# \        V \        4      '       d8   V P                  4        UUu/ uF  w  r!\        V4      \        V4      bK  	  upp# V # u upi u uppi )z6Convert telemetry values into log-friendly primitives.)
Ú
isinstanceÚtorchr'   ÚstrÚTensorÚlistÚtupleÚ_normalize_fieldÚdictÚitems)r5   ÚvÚks   &  r   r=   r=   D   s±   € ô �%œŸ™×&Ò&Ü�5‹zÐÜ�%œŸ™×&Ò&Ü�5—<‘<Ó Ð Ü�%œ$¤˜×'Ò'Ù-2Ó3©U¨Ô  Ö#©UÑ3Ð3Ü�%œ×ÒØ8=¿¹¼ÔF¹±°”�A“Ô(¨Ó+Ò+¹ÒFÐFØ€Lùò 4ùãFs   ÂC'Ã C,c               ó$   € V ^8„  d   QhRRRRRR/# )r   Úeventr9   Úfieldsr   r   ÚNoner   )r   s   "r   r   r   R   s!   € ÷ +ñ + ð +°ð +¸ñ +r   c                ój  € \        4       '       g   R# RV R\        \        P                  ! 4       ^4      /pVP                  4        F  w  r4\	        V4      W#&   K  	  \
        ;_uu_ 4        \        P                  V4       RRR4       \        P                  RV 24       R#   + '       g   i     L*; i)z;Record and log one structured telemetry event when enabled.NrC   ÚtszDeviceTelemetry: )
r   ÚroundÚtimer?   r=   Ú_records_lockr   ÚappendÚlogÚinfo)rC   rD   ÚrecordÚkeyr5   s   &,   r   Úemit_device_telemetryrP   R   s†   € ô $×%Ò%Ùð 	�ØŒe”D—I’I“K Ó#ð€Fð —l‘l–n‰
ˆÜ& uÓ-ˆ‹ñ %÷ 
ŽÜ�‰˜Ô÷ 
ô ‡H�HÐ   Ð)Ö*÷ 
�ús   Á*B"Â"B2	c               ó   € V ^8„  d   QhRR/# )r   r   rE   r   )r   s   "r   r   r   e   s   € ÷ ñ ¨ñ r   c                 ó„   € \         ;_uu_ 4        \        P                  4        RRR4       R#   + '       g   i     R# ; i)z.Discard previously captured telemetry records.N)rJ   r   Úclearr   r   r   Úclear_device_telemetry_recordsrT   e   s   € ÷ 
ŽÜ�‰Ô÷ 
��Šús   �.®?	c               ó   € V ^8„  d   QhRR/# )r   r   r   r   )r   s   "r   r   r   l   s   € ÷ 5ñ 5Ð&:ñ 5r   c                 ó¤   € \         ;_uu_ 4        \         U u. uF  p \        V 4      NK  	  up uuRRR4       # u up i   + '       g   i     R# ; i)z0Return a copy of the captured telemetry records.N)rJ   r   r>   )rN   s    r   Úget_device_telemetry_recordsrW   l   s3   € ÷ 
ŽÝ+3Ó4©8 ”�V–©8Ñ4÷ 
ŠùÚ4÷ 
�ˆús   �	>˜9¬>¹>¾A	)r3   rT   r   r"   rP   rW   r/   ) Ú__conditional_annotations__Ú__doc__Ú
__future__r   Ú	threadingrI   Ú
contextlibr   Úcontextvarsr   Ú	functoolsr   Útypingr   r8   Úloggerr   rL   r   ÚLockrJ   r   Ú__annotations__r   r"   r/   r3   r=   rP   rT   rW   Ú__all__)rX   s   @r   Ú<module>rd      s�   øðô Iå "ã Û Ý %Ý "Ý Ý ã å  áƒn€áÐ2¸EÔB€Ø—’Ó €Ø!#€Ð
Ó #õð ôó ðòò	õõ+õ&õ5ò‚r   