From ad54c5992eda08520997db8ec488eb78d7542809 Mon Sep 17 00:00:00 2001 From: Marko Radosavljevic Date: Wed, 15 Jul 2026 10:27:15 -0400 Subject: [PATCH] Reset dynamo state in test_memory_compile_regions --- test/test_cuda.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/test_cuda.py b/test/test_cuda.py index b425c036597e6..07489c9f1cfc1 100644 --- a/test/test_cuda.py +++ b/test/test_cuda.py @@ -4653,6 +4653,7 @@ def forward(self, x): return x try: + torch._dynamo.reset() torch.cuda.memory.empty_cache() input_tensor = torch.randn(1, 10, device="cuda") # Create an instance of the model