Ë
    g^(ht  ã                   ó�   — 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y)é    N)ÚUnion)Ú	run_testsÚTestCase)Úconfig)Úfresh_inductor_cacheÚneeds.Úreturnc                 ó   — t        | «       y ©N)Údynamo_run_tests)r   s    úW/var/www/skyplay_api_hub/venv/lib/python3.12/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.
    c                 ó
  •— t         ‰| �  «        t        j                  «       | _        | j                  j                  t        j                  ddi«      «       dt        j                  vr0| 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Ú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Ñ$8Ó$:ˆÔ!Ø×!Ñ!×/Ñ/Ü×"Ñ"à+¨Tðóô	
ð *´·±Ñ;Ø×%Ñ%×3Ñ3Ü—‘Ð.°Ð5Ó6ôô
 �J‰J�N‰NÐ>Ó?À3ÒFÜ—
‘
—‘Ð4Ó5¸Ò<à×%Ñ%×3Ñ3Ô4HÓ4JÕKð =ð Gr   c                 óV   •— t         ‰| �  «        | j                  j                  «        y r   )r   ÚtearDownr   Úcloser"   s    €r   r&   zTestCase.tearDown.   s    ø€ Ü‰ÑÔØ×!Ñ!×'Ñ'Õ)r   )r	   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r&   Ú__classcell__)r$   s   @r   r   r      s   ø„ ñõ
L÷,*ñ *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      sO   ðÛ Û 	Ý ÷õ 8Ý "Ý 6ñ�U˜3  c¨3 h¡Ð/Ñ0ð ¸$ó ô*ˆ~õ *r   