frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[dynamo] Mark a vt unspecialized nn module variable source earlier (#154780)"

Browse Source 
This reverts commit cc96febb979da16b0a0b758020b330a49c72b7e7.

Reverted https://github.com/pytorch/pytorch/pull/154780 on behalf of https://github.com/seemethere due to This fails internal testing see, https://fburl.com/diff/b0yuxk4w ([comment](https://github.com/pytorch/pytorch/pull/154780#issuecomment-2940381691))
This commit is contained in:
PyTorch MergeBot
2025-06-04 15:03:34 +00:00
parent a0f2544502
commit a99a01a677
8 changed files with 19 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/dynamo/test_modules.py
View File

@ -1299,7 +1299,6 @@ class NNModuleTests(torch._dynamo.test_case.TestCase):
self.assertTrue(torch._dynamo.testing.same(r, m(i)))
self.assertEqual(cnt.op_count, 6)
@patch.object(torch._dynamo.config, "allow_unspec_int_on_nn_module", True)
def test_self_mutating1(self):
m1 = torch.nn.Linear(10, 10)
m2 = SelfMutatingModule(m1)