o
    wZŽ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ÚtorchZ_CZ_jit_to_static_moduler   Ústatic_moduleÚgraph)ÚselfZscripted© r	   úT/var/www/auris/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   Zwarmup_runsZ	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   