o
    ‡ZŽht  ã                   @   sˆ   d dl Z d dlZd dlmZ d dlmZmZ d dl	m
Z d dlm
Z
 d dlmZ ddeeeedf f d	dfd
d„ZG dd„ deƒZdS )é    N)ÚUnion)Ú	run_testsÚTestCase)Úconfig)Úfresh_inductor_cache© Úneeds.Úreturnc                 C   s   t | ƒ d S ©N)Údynamo_run_tests)r   r   r   úH/var/www/auris/lib/python3.10/site-packages/torch/_inductor/test_case.pyr      s   r   c                       s0   e Zd ZdZd‡ fdd„Zd‡ fdd„Z‡  ZS )	r   zz
    A base TestCase for inductor tests. Enables FX graph caching and isolates
    the cache directory for each test.
    r	   Nc                    s„   t ƒ  ¡  t ¡ | _| j t ddi¡¡ dtj	vr%| j t
 ddi¡¡ tj	 d¡dkr>tj	 d¡dkr@| j tƒ ¡ d S d S d S )NZenable_autograd_cacheTZTORCHINDUCTOR_FX_GRAPH_CACHEZfx_graph_cacheZ!INDUCTOR_TEST_DISABLE_FRESH_CACHEÚ1ZTORCH_COMPILE_DEBUG)ÚsuperÚsetUpÚ
contextlibÚ	ExitStackÚ_inductor_test_stackÚenter_contextÚfunctorch_configÚpatchÚosÚenvironr   Úgetr   ©Úself©Ú	__class__r   r   r      s"   

ÿÿÿ
ÿýzTestCase.setUpc                    s   t ƒ  ¡  | j ¡  d S r
   )r   ÚtearDownr   Úcloser   r   r   r   r   .   s   
zTestCase.tearDown)r	   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   Ú__classcell__r   r   r   r   r      s    r   )r   )r   r   Útypingr   Ztorch._dynamo.test_caser   r   r   ZDynamoTestCaseZtorch._functorchr   r   Ztorch._inductorZtorch._inductor.utilsr   ÚstrÚtupler   r   r   r   Ú<module>   s    $