deprecate

pytorch_lightning/plugins/training_type/training_type_plugin.py

@@ -464,3 +464,10 @@ def pre_dispatch(self, trainer: "pl.Trainer") -> None:
     def dispatch(self, trainer: "pl.Trainer") -> None:
         """Hook to do something before the training/evaluation/prediction starts."""
         self.precision_plugin.dispatch(trainer)
+
+    def post_dispatch(self, trainer: "pl.Trainer") -> None:
+        """Hook to do something after the training/evaluation/prediction finishes.
+
+        .. deprecated:: v1.6
+            This method has been deprecated in v1.6 and will be removed in v1.7. Use :meth:`teardown` instead.
+        """

pytorch_lightning/trainer/trainer.py

@@ -1224,6 +1224,12 @@ def _log_hyperparams(self) -> None:
             self.logger.save()
 
     def _post_dispatch(self):
+        if is_overridden("post_dispatch", self.training_type_plugin, parent=TrainingTypePlugin):
+            rank_zero_deprecation(
+                f"`{self.training_type_plugin.__class__.__name__}.post_dispatch()` has been deprecated in v1.6 and"
+                " will be removed in v1.7. Move your implementation to"
+                f" `{self.training_type_plugin.__class__.__name__}.teardown()` instead."
+            )
         # these `teardown` calls are here instead of in `_call_teardown_hook` since they are internal teardowns
         # which need to happen before.
         self.accelerator.teardown()