
    7hb                         S SK r S SKrS SKJ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   4   SS4S	 jjr " S
 S\5      rg)    N)Union)	run_testsTestCase)config)fresh_cacheneeds.returnc                     [        U 5        g N)dynamo_run_tests)r   s    S/var/www/fran/franai/venv/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                  ;  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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	
 *;%%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.tearDown.   s     !!'')r   )r   )r	   N)	__name__
__module____qualname____firstlineno____doc__r   r(   __static_attributes____classcell__)r%   s   @r   r   r      s    
C,* *r   r   ) )r   r    typingr   torch._dynamo.test_caser   r   r   DynamoTestCasetorch._functorchr   r   torch._inductortorch._inductor.utilsr   strtupler2   r   r   <module>r;      sO     	  8 " -U3c3h/0 $ *~ *r   