
    [Th                         S SK Jr  S SKrS SKJr  S SKJr  S SKJr  S/r	 " S S\R                  R                  R                  5      r " S	 S\5      rg)
    )CallableN)_FusedModule)Tracer)ScopeQuantizationTracerc                   `   ^  \ rS rSrS\S\R                  R                  S\4U 4S jjr	Sr
U =r$ )ScopeContextManager   scopecurrent_modulecurrent_module_pathc           	      L   > [         TU ]  U[        U[        U5      5      5        g )N)super__init__r   type)selfr   r   r   	__class__s       W/var/www/auris/envauris/lib/python3.13/site-packages/torch/ao/quantization/fx/tracer.pyr   ScopeContextManager.__init__   s!     	&94;O PQ     )__name__
__module____qualname____firstlineno__r   torchnnModulestrr   __static_attributes____classcell__r   s   @r   r	   r	      s2    RR,1HHOORRUR Rr   r	   c                   ~   ^  \ rS rSrS\\   S\\   4U 4S jjrS\R                  R                  S\S\4S jrS	rU =r$ )
r      skipped_module_namesskipped_module_classesc                 j   > [         TU ]  5         Xl        X l        [	        SS 5      U l        SU l        g )N T)r   r   r%   r&   r   r   record_stack_traces)r   r%   r&   r   s      r   r   QuantizationTracer.__init__   s3     	$8!&<# 2t_
#' r   mmodule_qualified_namereturnc                 r   UR                   R                  S5      =(       d    UR                   R                  S5      =(       a)    [        U[        R                  R
                  5      (       + =(       dD    X R                  ;   =(       d/    [        U5      U R                  ;   =(       d    [        U[        5      $ )Nztorch.nnztorch.ao.nn)
r   
startswith
isinstancer   r   
Sequentialr%   r   r&   r   )r   r+   r,   s      r   is_leaf_module!QuantizationTracer.is_leaf_module$   s     LL++J7 >||..}=; #1ehh&9&9::	+ %(A(AA	+ Aw$555	+ !\*	
r   )r)   r   r&   r%   )r   r   r   r   listr   r   r   r   r   r   boolr2   r    r!   r"   s   @r   r   r      sH    ($(I(GKH~(
 
 
PT 
 
r   )typingr   r   torch.ao.nn.intrinsicr   torch.fx._symbolic_tracer   torch.fx.proxyr   __all__fxproxyr	   r   r   r   r   <module>r=      sH      . +   
R%((..<< R
 
r   