SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[tests] reset logs in torch.compile test (#37894)

Browse Source
This commit is contained in:
Joao Gante
2025-04-30 16:04:28 +01:00
committed by GitHub
parent 1b222903c3
commit 8e8025b384
1 changed files with 21 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/generation/test_utils.py
View File

@@ -2175,6 +2175,8 @@ class GenerationTesterMixin:
compiled_outputs = [] compiled_outputs = []
# Uses a context manager to catch recompilation logs. If there is any recompilation, this test fails. # Uses a context manager to catch recompilation logs. If there is any recompilation, this test fails.
# Try/Finally is used to ensure that the log options are reset even if an error is raised.
try:
torch._logging.set_logs(recompiles_verbose=True) torch._logging.set_logs(recompiles_verbose=True)
logger = logging.get_logger("torch._dynamo.guards") logger = logging.get_logger("torch._dynamo.guards")
with CaptureLogger(logger) as cl: with CaptureLogger(logger) as cl:
@@ -2192,6 +2194,8 @@ class GenerationTesterMixin:
self.assertTrue(decoder_cache.is_compileable) self.assertTrue(decoder_cache.is_compileable)
# our auto compile should have been called # our auto compile should have been called
self.assertTrue(hasattr(model_to_be_compiled, "_compiled_call")) self.assertTrue(hasattr(model_to_be_compiled, "_compiled_call"))
finally:
torch._logging.set_logs()
if "Recompiling" in cl.out or ("guard" in cl.out and "failure" in cl.out): if "Recompiling" in cl.out or ("guard" in cl.out and "failure" in cl.out):
raise RuntimeError( raise RuntimeError(