o h@srddlZddlZddlZddlZddlZddlmZmZmZm Z ddl m Z m Z mZd ddZGdd d e Z dS) N) IS_WINDOWSTEST_WITH_CROSSREFTEST_WITH_TORCHDYNAMOTestCase)configresetutilsc Csddlm}tstststjdkrdSt|tr|f}|D] }|dkr+t j s+dSzt |Wqty=YdSw|dS)Nr) run_tests) cuda)$torch.testing._internal.common_utilsr rrrsys version_info isinstancestrtorchr is_available importlib import_module ImportError)needsr needr r M/var/www/html/ai/venv/lib/python3.10/site-packages/torch/_dynamo/test_case.pyr s(     r csHeZdZefddZefddZfddZfddZZS) rcs|jtdSN) _exit_stackclosesuper tearDownClasscls __class__r rr *s zTestCase.tearDownClasscs.tt|_|jtjddddS)NTF)raise_on_ctx_manager_usagesuppress_errors)r setUpClass contextlib ExitStackr enter_contextrpatchr!r#r rr'/s   zTestCase.setUpClasscstttjdSr)rsetUprr countersclear)selfr#r rr,7s zTestCase.setUpcs@tjD] \}}t||qttjtdSr) r r-itemsprint most_commonrr.rtearDown)r/kvr#r rr3<s  zTestCase.tearDown) __name__ __module__ __qualname__ classmethodr r'r,r3 __classcell__r r r#rr)s r)r )r(rrr torch.testingrrrrr TorchTestCaserrr r r r r rs