
    `i}                     (    d dl Z  G d d          ZdS )    Nc                   ,    e Zd Zd Zd Zd Zd Zd ZdS )StaticModulec                     t          |d          r+t          j                            |j                  | _        d S t          j                            |j                  | _        d S )N_c)hasattrtorch_C_jit_to_static_moduler   static_modulegraph)selfscripteds     y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/testing/_internal/static_module.py__init__zStaticModule.__init__   sU    8T"" 	P!&!?!?!L!LD!&!?!?!O!OD    c                      | j         |i |S N)r   r   argskwargss      r   __call__zStaticModule.__call__   s    !t!426222r   c                 @    | j                             ||||           d S r   )r   	benchmarkr   r   r   warmup_runs	main_runss        r   r   zStaticModule.benchmark   s%    $$T6;	JJJJJr   c                 8    | j                             ||          S r   )r   runAsyncr   s      r   r   zStaticModule.runAsync   s    !**4888r   c                 <    | j                             ||||          S r   )r   benchmark_individual_opsr   s        r   r    z%StaticModule.benchmark_individual_ops   s&    !::&+y
 
 	
r   N)__name__
__module____qualname__r   r   r   r   r     r   r   r   r      se        P P P3 3 3K K K9 9 9
 
 
 
 
r   r   )r   r   r$   r   r   <module>r%      sB    
 
 
 
 
 
 
 
 
 
r   