
    ɯwg                     Z    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Z G d de      Zy)	    N)	run_testsTestCase)config)fresh_inductor_cachec                     t        |        y N)dynamo_run_tests)needss    ^/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/torch/_inductor/test_case.pyr   r      s
    U    c                   ,     e Zd ZdZ fdZ 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.
    c                    t         |           t        j                         | _        | j                  j                  t        j                  ddi             t        j                  j                  d      dk7  rGt        j                  j                  d      dk7  r$| j                  j                  t                      y y y )Nfx_graph_cacheT!INDUCTOR_TEST_DISABLE_FRESH_CACHE1TORCH_COMPILE_DEBUG)supersetUp
contextlib	ExitStack_inductor_test_stackenter_contextr   patchosenvirongetr   self	__class__s    r   r   zTestCase.setUp   s    $.$8$8$:!!!//>NPT=U0VWJJNN>?3F

45<%%334H4JK = Gr   c                 V    t         |           | j                  j                          y r   )r   tearDownr   closer   s    r   r!   zTestCase.tearDown!   s     !!'')r   )__name__
__module____qualname____doc__r   r!   __classcell__)r   s   @r   r   r      s    
L* *r   r   ) )r   r   torch._dynamo.test_caser   r	   r   DynamoTestCasetorch._inductorr   torch._inductor.utilsr   r(   r   r   <module>r-      s)     	 # 6*~ *r   