
    `ib                         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
Z G d de          ZdS )    N)Union)	run_testsTestCase)config)fresh_cache needs.returnc                 $    t          |            d S N)dynamo_run_tests)r	   s    m/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/_inductor/test_case.pyr   r      s    U    c                   0     e Zd ZdZd fdZd fdZ xZS )r   zz
    A base TestCase for inductor tests. Enables FX graph caching and isolates
    the cache directory for each test.
    r
   Nc                 ,   t                                                       t          j                    | _        | j                            t          j        ddi                     dt          j	        vr.| j                            t          j        ddi                     t          j	                            d          dk    rKt          j	                            d          dk    r*| j                            t                                 d S d S d S )Nenable_autograd_cacheTTORCHINDUCTOR_FX_GRAPH_CACHEfx_graph_cache!INDUCTOR_TEST_DISABLE_FRESH_CACHE1TORCH_COMPILE_DEBUG)supersetUp
contextlib	ExitStack_inductor_test_stackenter_contextfunctorch_configpatchosenvironr   getr   self	__class__s    r   r   zTestCase.setUp   s    $.$8$:$:!!//"+T 	
 	
 	
 *;;%33.566  
 JNN>??3FF
455<<%33KMMBBBBB GF<<r   c                 z    t                                                       | j                                         d S r   )r   tearDownr   closer#   s    r   r'   zTestCase.tearDown.   s4    !'')))))r   )r
   N)__name__
__module____qualname____doc__r   r'   __classcell__)r%   s   @r   r   r      sk         
C C C C C C,* * * * * * * * * *r   r   )r   )r   r    typingr   torch._dynamo.test_caser   r   r   DynamoTestCasetorch._functorchr   r   torch._inductortorch._inductor.utilsr   strtupler   r   r   <module>r6      s        				              8 7 7 7 7 7 " " " " " " - - - - - - U3c3h/0 $    * * * * *~ * * * * *r   