[EASY][TEST][CUDA] Fix typo in test_graph_make_graphed_callables_same_pool (#132059)

Per title.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/132059
Approved by: https://github.com/Skylion007
This commit is contained in:
Aidyn-A 2024-07-29 19:15:35 +00:00 committed by PyTorch MergeBot
parent 5cc34f61d1
commit 301ec32ae8

View file

@ -3402,7 +3402,7 @@ exit(2)
self.assertEqual(p, pg)
self.assertEqual(p.grad, pg.grad)
self.assertNotEqual(p.data_ptr(), pg.data_ptr())
self.assertNotEqual(p.grad.data_ptr, pg.grad.data_ptr)
self.assertNotEqual(p.grad.data_ptr(), pg.grad.data_ptr())
def _test_graphed_optimizer(
self, steps_warmup, steps_train, optimizer_ctor, kwargs