enable optim tests on dynamo to test flaky bot (#86976)
will link the issue that disabled them if this gets approved
Pull Request resolved: https://github.com/pytorch/pytorch/pull/86976
Approved by: https://github.com/albanD
diff --git a/test/test_optim.py b/test/test_optim.py
index ac255b8..c45b2ac 100644
--- a/test/test_optim.py
+++ b/test/test_optim.py
@@ -19,7 +19,7 @@
EPOCH_DEPRECATION_WARNING
from torch.optim.swa_utils import AveragedModel, SWALR, update_bn
from torch.testing._internal.common_utils import TestCase, run_tests, TEST_WITH_UBSAN, load_tests, \
- parametrize, instantiate_parametrized_tests, gradcheck, skipIfRocm, skipIfTorchDynamo
+ parametrize, instantiate_parametrized_tests, gradcheck, skipIfRocm
# load_tests from common_utils is used to automatically filter tests for
# sharding on sandcastle. This line silences flake warnings
load_tests = load_tests
@@ -34,7 +34,7 @@
x, y = tensor
return torch.tensor((-400 * x * (y - x ** 2) - 2 * (1 - x), 200 * (y - x ** 2)))
-@skipIfTorchDynamo()
+
class TestOptim(TestCase):
exact_dtype = True