
    sh}                          S SK r  " S S5      rg)    Nc                   2    \ rS rSrS rS rS rS rS rSr	g)	StaticModule   c                     [        US5      (       a/  [        R                  R                  UR                  5      U l        g [        R                  R                  UR                  5      U l        g )N_c)hasattrtorch_C_jit_to_static_moduler   static_modulegraph)selfscripteds     y/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/testing/_internal/static_module.py__init__StaticModule.__init__   sE    8T""!&!?!?!LD!&!?!?!OD    c                 &    U R                   " U0 UD6$ Nr   r   argskwargss      r   __call__StaticModule.__call__   s    !!42622r   c                 <    U R                   R                  XX45        g r   )r   	benchmarkr   r   r   warmup_runs	main_runss        r   r   StaticModule.benchmark   s    $$T;Jr   c                 8    U R                   R                  X5      $ r   )r   runAsyncr   s      r   r#   StaticModule.runAsync   s    !!**488r   c                 :    U R                   R                  XX45      $ r   )r   benchmark_individual_opsr   s        r   r&   %StaticModule.benchmark_individual_ops   s     !!::+
 	
r   r   N)
__name__
__module____qualname____firstlineno__r   r   r   r#   r&   __static_attributes__ r   r   r   r      s    P3K9
r   r   )r	   r   r-   r   r   <module>r.      s    
 
r   