
    3j                         S SK r S SKrS SKJrJr  S SKJr	  S SK
Jr  S SKJr  SS\\\S4   -  SS4S jjr " S	 S
\5      rg)    N)	run_testsTestCase)config)fresh_cacheneeds.returnc                     [        U 5        g N)dynamo_run_tests)r   s    S/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/_inductor/test_case.pyr   r      s
    U    c                   @   ^  \ rS rSrSrSU 4S jjrSU 4S jjrSrU =r$ )r      zn
A base TestCase for inductor tests. Enables FX graph caching and isolates
the cache directory for each test.
c                 @  > [         TU ]  5         [        R                  " 5       U l        U R                  R                  [        R                  " SS05      5        S[        R                  ;  aE  S[        R                  ;  a1  U R                  R                  [        R                  " SS05      5        [        R                  R                  S5      S:w  aH  [        R                  R                  S5      S:w  a$  U R                  R                  [        5       5        g g g )	Nenable_autograd_cacheTTORCHINDUCTOR_FX_GRAPH_CACHE$TORCHINDUCTOR_FX_GRAPH_CACHE_DEFAULT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   TestCase.setUp   s    $.$8$8$:!!!//""+T	
 +"**<6bjjH%%33.56
 JJNN>?3F

45<%%33KMB = Gr   c                 V   > [         TU ]  5         U R                  R                  5         g r
   )r   tearDownr   closer#   s    r   r(   TestCase.tearDown0   s     !!'')r   )r   )r   N)	__name__
__module____qualname____firstlineno____doc__r   r(   __static_attributes____classcell__)r%   s   @r   r   r      s    
C2* *r   r   ) )r   r    torch._dynamo.test_caser   r   r   DynamoTestCasetorch._functorchr   r   torch._inductortorch._inductor.utilsr   strtupler2   r   r   <module>r:      sG     	 8 " -S5c?* D !*~ !*r   