o
    0ñh}  ã                   @   s   d dl Z G dd„ dƒZdS )é    Nc                   @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )ÚStaticModulec                 C   s2   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Úscripted© r   úY/var/www/vscode/kcb/lib/python3.10/site-packages/torch/testing/_internal/static_module.pyÚ__init__   s   
zStaticModule.__init__c                 O   s   | j |i |¤ŽS ©N)r   ©r
   ÚargsÚkwargsr   r   r   Ú__call__   s   zStaticModule.__call__c                 C   s   | j  ||||¡ d S r   )r   Ú	benchmark©r
   r   r   Úwarmup_runsÚ	main_runsr   r   r   r      s   zStaticModule.benchmarkc                 C   s   | j  ||¡S r   )r   ÚrunAsyncr   r   r   r   r      s   zStaticModule.runAsyncc                 C   s   | j  ||||¡S r   )r   Úbenchmark_individual_opsr   r   r   r   r      s   ÿz%StaticModule.benchmark_individual_opsN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   r   r   r      s    r   )r   r   r   r   r   r   Ú<module>   s   