From 4c0894dc05d09a1973444c5d7db1ca11e9f7fd92 Mon Sep 17 00:00:00 2001 From: soulitzer Date: Mon, 12 May 2025 16:34:52 -0400 Subject: [PATCH 1/3] Make test_create_graph_and_full_backward_hook_cycle more robust to random warnings [ghstack-poisoned] --- test/test_autograd.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/test/test_autograd.py b/test/test_autograd.py index 4b2f937067f57..5cb70f1aa07c7 100644 --- a/test/test_autograd.py +++ b/test/test_autograd.py @@ -9335,10 +9335,9 @@ def forward(self, x): with set_warn_always_context(True): with warnings.catch_warnings(record=True) as w: tmp.exp().sum().backward(create_graph=True) + filtered_warnings = [ + w_ for w_ in w if "Using backward() with create_graph=True" in str(w_)] self.assertTrue(len(w) == 1) - self.assertTrue( - "Using backward() with create_graph=True" in str(w[0].message) - ) # Remove the backward + create_graph=True cycle a.grad = None From e4babddb220595edb2bab5acc2c1ebee4e46a0de Mon Sep 17 00:00:00 2001 From: soulitzer Date: Mon, 12 May 2025 16:40:37 -0400 Subject: [PATCH 2/3] Update on "Make test_create_graph_and_full_backward_hook_cycle more robust to unrelated warnings" [ghstack-poisoned] --- test/test_autograd.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/test_autograd.py b/test/test_autograd.py index 5cb70f1aa07c7..95858ecca1711 100644 --- a/test/test_autograd.py +++ b/test/test_autograd.py @@ -9333,11 +9333,11 @@ def forward(self, x): tmp.grad_fn.metadata["a"] = test with set_warn_always_context(True): - with warnings.catch_warnings(record=True) as w: + with warnings.catch_warnings(record=True) as ws: tmp.exp().sum().backward(create_graph=True) filtered_warnings = [ - w_ for w_ in w if "Using backward() with create_graph=True" in str(w_)] - self.assertTrue(len(w) == 1) + w for w in ws if "Using backward() with create_graph=True" in str(w.message)] + self.assertTrue(len(ws) == 1) # Remove the backward + create_graph=True cycle a.grad = None From 0c166fc0924bf700c828eb0470f5ae89769e6a88 Mon Sep 17 00:00:00 2001 From: soulitzer Date: Mon, 12 May 2025 16:53:58 -0400 Subject: [PATCH 3/3] Update on "Make test_create_graph_and_full_backward_hook_cycle more robust to unrelated warnings" [ghstack-poisoned] --- test/test_autograd.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_autograd.py b/test/test_autograd.py index 95858ecca1711..ff125f0014ea1 100644 --- a/test/test_autograd.py +++ b/test/test_autograd.py @@ -9335,9 +9335,9 @@ def forward(self, x): with set_warn_always_context(True): with warnings.catch_warnings(record=True) as ws: tmp.exp().sum().backward(create_graph=True) - filtered_warnings = [ + filtered_ws = [ w for w in ws if "Using backward() with create_graph=True" in str(w.message)] - self.assertTrue(len(ws) == 1) + self.assertTrue(len(filtered_ws) == 1) # Remove the backward + create_graph=True cycle a.grad = None