re-enable fsdp1 dynamo multi-gpu test after upstream bug fix

Author: speediedan

tests/test_fsdp.py

@@ -30,7 +30,7 @@
 from tests.helpers.common import get_fts, unexpected_warns, unmatched_warns, nones
 from tests.helpers.runif import RunIf, RUNIF_MAP
 from tests.helpers.expected_warns import FSDP_BASE_WARNS, FSDP_DYNAMO_EXPECTED_WARNS
-from tests.fsdp_expected_paths import (path_default, path_default_orig, path_ignore_p_uo,
+from tests.fsdp_expected_paths import (path_default, path_default_orig, path_ignore_p_uo, path_default_orig_eo_dyn,
                                        path_8_14, path_5_10, path_ext_7_14, path_ext_8_16, path_optimlr_reinit,
                                        lrs_path_optimlr_reinit, path_bn_track_false, path_bn_track_true, ResultEnum)
 
@@ -642,11 +642,11 @@ def warn_custom_auto_wrap_policy(
         None,
         (path_default_orig, "degenerate and unintended", *nones(2)),
     ),
-    # "cust_awp_noprec_dynamo": (
-    #     (nond_loss_adam_model, cust_awp, False, 7, unwrap_7_dyn, None, epoch_t_only, max_epoch_4, None),
-    #     None,
-    #     (path_default_orig_eo_dyn, *nones(3)),
-    # ),
+    "cust_awp_noprec_dynamo": (
+        (nond_loss_adam_model, cust_awp, False, 7, unwrap_7_dyn, None, epoch_t_only, max_epoch_4, None),
+        None,
+        (path_default_orig_eo_dyn, *nones(3)),
+    ),
     "cust_awp_mwp_reinitlr_optim_no_use_orig": (
         (base_model, awp_mwp_parity, True, 8, unwrap_7_mp, None, opt_inspect, None, DISABLE_USE_ORIG),
         None,