
    sg}                          d dl Z  G d d      Zy)    Nc                   *    e Zd Zd Zd Zd Zd Zd Zy)StaticModulec                     t        |d      r/t        j                  j                  |j                        | _        y t        j                  j                  |j                        | _        y )N_c)hasattrtorch_C_jit_to_static_moduler   static_modulegraph)selfscripteds     X/var/www/html/venv/lib/python3.12/site-packages/torch/testing/_internal/static_module.py__init__zStaticModule.__init__   sB    8T"!&!?!?!LD!&!?!?!OD    c                 &     | j                   |i |S N)r   r   argskwargss      r   __call__zStaticModule.__call__   s    !t!!42622r   c                 @    | j                   j                  ||||       y r   )r   	benchmarkr   r   r   warmup_runs	main_runss        r   r   zStaticModule.benchmark   s    $$T6;	Jr   c                 :    | j                   j                  ||      S r   )r   runAsyncr   s      r   r   zStaticModule.runAsync   s    !!**488r   c                 >    | j                   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      s    P3K9
r   r   )r   r   r$   r   r   <module>r%      s    
 
r   