U
    yh                     @   sV   d dl Z d dlZd dlmZmZ d dlmZ d dl	m
Z
 d
ddZG dd	 d	eZdS )    N)	run_testsTestCase)config)fresh_inductor_cache c                 C   s   t |  d S N)dynamo_run_tests)Zneedsr   r   K/var/www/html/venv/lib/python3.8/site-packages/torch/_inductor/test_case.pyr      s    r   c                       s,   e Zd ZdZ fddZ 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.
    c                    s\   t    t | _| jtddi tj	
ddkrXtj	
ddkrX| jt  d S )NZfx_graph_cacheTZ!INDUCTOR_TEST_DISABLE_FRESH_CACHE1ZTORCH_COMPILE_DEBUG)supersetUp
contextlib	ExitStack_inductor_test_stackenter_contextr   patchosenviron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)__name__
__module____qualname____doc__r   r   __classcell__r   r   r   r	   r      s   
r   )r   )r   r   Ztorch._dynamo.test_caser   r   r   ZDynamoTestCaseZtorch._inductorr   Ztorch._inductor.utilsr   r   r   r   r	   <module>   s   
