+
    –°³jÁ  ã                   óÄ   € ^ RI t ^ RIHt ^ RIt^ RIHt ^ RIHt . ROt ! R	 R]4      t	 ! R
 R]4      t
R R ltR R lt ! R R]
4      tR R ltR t ! R R]4      tR# )é    N)ÚAny)ÚGraphModule)ÚGraphÚFusedGraphModuleÚObservedGraphModuleÚObservedStandaloneGraphModuleÚQuantizedGraphModulec                   óD   a a€ ] tR t^t oV3R lV 3R lltR tRtVtV ;t# )r   c                óŠ   <€ V ^8„  d   QhRS[ P                  P                  S[S[S[3,          ,          RS[RS[S[,          /# ©é   ÚrootÚgraphÚpreserved_attr_names©ÚtorchÚnnÚModuleÚdictÚstrr   r   Úset)ÚformatÚ__classdict__s   "€Úf/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/ao/quantization/fx/graph_module.pyÚ__annotate__ÚFusedGraphModule.__annotate__   sB   ø€ ÷ 7ñ 7á�h‰h�o‰o¡¡S©# X¥Õ.ð7ñ ð7ñ "¡#�hñ	7ó    c                óâ   <€ W0n         V P                    Uu/ uF"  p\        W4      '       g   K  V\        W4      bK$  	  pp\        SV `  W4       V F  p\        WWT,          4       K  	  R # u upi ©N)r   ÚhasattrÚgetattrÚsuperÚ__init__Úsetattr©Úselfr   r   r   ÚattrÚpreserved_attrsÚ	__class__s   &&&&  €r   r#   ÚFusedGraphModule.__init__   sr   ø€ ð %9Ô!ð ×1Ò1ó
á1�Ü�t×"ô &ˆD”'˜$Ó%Ò%Ù1ð 	ð 
ô
 	‰Ñ˜Ô%Û#ˆDÜ�D Õ 5Ö6ó $ùò
s
   –A,®A,c                ó  € \         P                  P                  4       p\        P                  ! V P
                  4      Vn        \        V\        P                  ! V P                  4      \        P                  ! V P                  4      4      # r   )	r   r   r   ÚcopyÚdeepcopyÚ__dict__r   r   r   ©r&   ÚmemoÚfake_mods   && r   Ú__deepcopy__ÚFusedGraphModule.__deepcopy__&   sY   € Ü—8‘8—?‘?Ó$ˆÜ ŸMšM¨$¯-©-Ó8ˆÔÜØÜ�MŠM˜$Ÿ*™*Ó%Ü�MŠM˜$×3Ñ3Ó4ó
ð 	
r   ©r   ©	Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r#   r2   Ú__static_attributes__Ú__classdictcell__Ú__classcell__©r)   r   s   @@r   r   r      s   ù‡ € ÷7ó 7÷&
ò 
r   c                   óD   a a€ ] tR t^0t oV3R lV 3R lltR tRtVtV ;t# )r   c                óŠ   <€ V ^8„  d   QhRS[ P                  P                  S[S[S[3,          ,          RS[RS[S[,          /# r   r   )r   r   s   "€r   r   Ú ObservedGraphModule.__annotate__1   sB   ø€ ÷ 7ñ 7á�h‰h�o‰o¡¡S©# X¥Õ.ð7ñ ð7ñ "¡#�hñ	7r   c                ó  <€ 0 RmP                  V4      V n        V P                   Uu/ uF"  p\        W4      '       g   K  V\        W4      bK$  	  pp\        SV `  W4       V F  p\        WWT,          4       K  	  R# u upi )Ú_activation_post_process_mapN>
   Ú_is_qatÚ	_patternsÚ_qconfig_mappingÚ_node_name_to_scopeÚ_observed_node_namesÚ_node_name_to_qconfigÚ_prepare_custom_configrB   Ú _activation_post_process_indexesÚ"_equalization_node_name_to_qconfig)Úunionr   r    r!   r"   r#   r$   r%   s   &&&&  €r   r#   ÚObservedGraphModule.__init__1   s„   ø€ ò%
÷ ‰%Ð$Ó
%ð 	Ô!ð ×1Ò1ó
á1�Ü�t×"ô &ˆD”'˜$Ó%Ò%Ù1ð 	ð 
ô
 	‰Ñ˜Ô%Û#ˆDÜ�D Õ 5Ö6ó $ùò
s   ¨A>Á A>c                ó  € \         P                  P                  4       p\        P                  ! V P
                  4      Vn        \        V\        P                  ! V P                  4      \        P                  ! V P                  4      4      # r   )	r   r   r   r,   r-   r.   r   r   r   r/   s   && r   r2   Ú ObservedGraphModule.__deepcopy__O   sY   € Ü—8‘8—?‘?Ó$ˆÜ ŸMšM¨$¯-©-Ó8ˆÔÜ"ØÜ�MŠM˜$Ÿ*™*Ó%Ü�MŠM˜$×3Ñ3Ó4ó
ð 	
r   r4   r5   r=   s   @@r   r   r   0   s   ù‡ € ÷7ó 7÷<
ò 
r   c                ó0   € V ^8„  d   QhR\         R\        /# ©r   ÚmoduleÚreturn©r   Úbool)r   s   "r   r   r   Y   s   € ÷ Uñ U¤ð U¬ñ Ur   c                 óH   € \        V R 4      ;'       d    RV P                  9   # )ÚmetaÚ_observed_graph_module_attrs)r    rW   ©rR   s   &r   Ú_is_observed_modulerZ   Y   s!   € Ü�6˜6Ó"×TÐTÐ'EÈÏÉÑ'TÐTr   c                óz   € V ^8„  d   QhR\         P                  P                  \        ,          R\        R\
        /# )r   ÚmodelÚ	attr_namerS   )r   r   r   r   r   r   )r   s   "r   r   r   ]   s1   € ÷ ñ Ü�8‰8�?‰?œ[Õ(ðÜ58ðäñr   c                 ó†   € \        V R 4      '       d/   RV P                  9   d   \        V P                  R,          V4      # R# )rW   rX   N)r    rW   r!   )r\   r]   s   &&r   Ú_get_observed_graph_module_attrr_   ]   s7   € ô ˆu�f×ÒÐ"@ÀEÇJÁJÔ"NÜ�u—z‘zÐ"@ÕAÀ9ÓMÐMÙr   c                   óD   a a€ ] tR t^et oV3R lV 3R lltR tRtVtV ;t# )r   c                óŠ   <€ V ^8„  d   QhRS[ P                  P                  S[S[S[3,          ,          RS[RS[S[,          /# r   r   )r   r   s   "€r   r   Ú*ObservedStandaloneGraphModule.__annotate__f   sB   ø€ ÷ <ñ <á�h‰h�o‰o¡¡S©# X¥Õ.ð<ñ ð<ñ "¡#�hñ	<r   c                óN   <€ VP                  R R04      p\        SV `	  WV4       R# )Ú'_standalone_module_input_quantized_idxsÚ(_standalone_module_output_quantized_idxsN)rL   r"   r#   )r&   r   r   r   r)   s   &&&&€r   r#   Ú&ObservedStandaloneGraphModule.__init__f   s3   ø€ ð  4×9Ñ9à9Ø:ðó 
Ðô 	‰Ñ˜Ð&:Ö;r   c                ó  € \         P                  P                  4       p\        P                  ! V P
                  4      Vn        \        V\        P                  ! V P                  4      \        P                  ! V P                  4      4      # r   )	r   r   r   r,   r-   r.   r   r   r   r/   s   && r   r2   Ú*ObservedStandaloneGraphModule.__deepcopy__t   sY   € Ü—8‘8—?‘?Ó$ˆÜ ŸMšM¨$¯-©-Ó8ˆÔÜ,ØÜ�MŠM˜$Ÿ*™*Ó%Ü�MŠM˜$×3Ñ3Ó4ó
ð 	
r   © r5   r=   s   @@r   r   r   e   s   ù‡ € ÷<ó <÷
ò 
r   c                ó0   € V ^8„  d   QhR\         R\        /# rQ   rT   )r   s   "r   r   r   ~   s   € ÷ ñ ¬3ð ´4ñ r   c                 ób   € \        V 4      ;'       d    V P                  R ,          P                  # )rX   )rZ   rW   Úis_observed_standalone_modulerY   s   &r   Ú_is_observed_standalone_modulerm   ~   s.   € ä˜FÓ#÷ 	Vð 	VØ�K‰KÐ6Õ7×UÑUðr   c                 óÐ   € \        V 4       FV  pR V9   g   K  \        \        W4      \        P                  P
                  4      '       g   KA  \        W4      pWQW$,           &   KX  	  R# ©Ú_packed_weightN)ÚdirÚ
isinstancer!   r   Ú_CÚScriptObject)r&   ÚdestinationÚprefixÚ	keep_varsr]   Úpacked_weights   &&&&  r   Ú_save_packed_weightry   …   sO   € Ü˜–Yˆ	Ø˜yÖ(¬ZÜ�DÓ$¤e§h¡h×&;Ñ&;÷.
ô .
ô $ DÓ4ˆMØ.;˜Õ*Ó+ó r   c                   óT   a a€ ] tR t^Žt oRtV3R lV 3R lltV 3R ltR tRtVt	V ;t
# )r	   zøThis class is created to make sure PackedParams
(e.g. LinearPackedParams, Conv2dPackedParams) to appear in state_dict
so that we can serialize and deserialize quantized graph module with
torch.save(m.state_dict()) and m.load_state_dict(state_dict)
c                óŠ   <€ V ^8„  d   QhRS[ P                  P                  S[S[S[3,          ,          RS[RS[S[,          /# r   r   )r   r   s   "€r   r   Ú!QuantizedGraphModule.__annotate__•   sB   ø€ ÷ <ñ <á�h‰h�o‰o¡¡S©# X¥Õ.ð<ñ ð<ñ "¡#�hñ	<r   c                ó  <€ W0n         V P                    Uu/ uF"  p\        W4      '       g   K  V\        W4      bK$  	  pp\        SV `  W4       V F  p\        WWT,          4       K  	  V P                  \        4       R # u upi r   )r   r    r!   r"   r#   r$   Ú_register_state_dict_hookry   r%   s   &&&&  €r   r#   ÚQuantizedGraphModule.__init__•   s�   ø€ ð %9Ô!ð ×1Ò1ó
á1�Ü�t×"ô &ˆD”'˜$Ó%Ò%Ù1ð 	ð 
ô
 	‰Ñ˜Ô%Û#ˆDÜ�D Õ 5Ö6ñ $à×&Ñ&Ô':Ö;ùò
s
   –B®Bc           	     óX  <€ . pV Fq  p	V	P                  R 4      '       g   K  \        W,          \        P                  P                  4      '       g   KN  \        W	W,          4       VP                  V	4       Ks  	  V F  p	VP                  V	4       K  	  \        S
V `%  VVVVVVV4       R# ro   )
Ú
startswithrr   r   rs   rt   r$   ÚappendÚpopr"   Ú_load_from_state_dict)r&   Ú
state_dictrv   Úlocal_metadataÚstrictÚmissing_keysÚunexpected_keysÚ
error_msgsÚattrs_to_popr]   r)   s   &&&&&&&&  €r   r„   Ú*QuantizedGraphModule._load_from_state_dict¦   s�   ø€ ð ˆÛ#ˆIØ×#Ñ#Ð$4×5Ô5¼*ØÕ%¤u§x¡x×'<Ñ'<÷;ô ;ô ˜¨Õ)>Ô?Ø×#Ñ# IÖ.ñ $ó &ˆIØ�N‰N˜9Ö%ñ &ô 	‰Ñ%ØØØØØØØö	
r   c                ó  € \         P                  P                  4       p\        P                  ! V P
                  4      Vn        \        V\        P                  ! V P                  4      \        P                  ! V P                  4      4      # r   )	r   r   r   r,   r-   r.   r	   r   r   r/   s   && r   r2   Ú!QuantizedGraphModule.__deepcopy__Æ   sY   € Ü—8‘8—?‘?Ó$ˆÜ ŸMšM¨$¯-©-Ó8ˆÔÜ#ØÜ�MŠM˜$Ÿ*™*Ó%Ü�MŠM˜$×3Ñ3Ó4ó
ð 	
r   r4   )r6   r7   r8   r9   Ú__doc__r#   r„   r2   r:   r;   r<   r=   s   @@r   r	   r	   Ž   s$   ù‡ € ñ÷<ó <õ"
÷@
ò 
r   )r   r   r   r	   )r,   Útypingr   r   Útorch.fxr   Útorch.fx.graphr   Ú__all__r   r   rZ   r_   r   rm   ry   r	   ri   r   r   Ú<module>r”      sb   ðã Ý ã Ý  Ý  ò€ô
�{ô 
ô<&
˜+ô &
õRUõô
Ð$7ô 
õ2ò<ô?
˜;ö ?
r   