Add deprecated autocast comments for doc

guangyey · guangyey · commit 4009f213e55f · 2024-05-14T10:14:01.000Z
ghstack-source-id: 81d22fb Pull Request resolved: #126062
diff --git a/torch/cpu/amp/autocast_mode.py b/torch/cpu/amp/autocast_mode.py
@@ -1,3 +1,4 @@
+import warnings
 from typing import Any
 
 import torch
@@ -8,7 +9,7 @@
 class autocast(torch.amp.autocast_mode.autocast):
     r"""
     See :class:`torch.autocast`.
-    ``torch.cpu.amp.autocast(args...)`` is equivalent to ``torch.autocast("cpu", args...)``
+    ``torch.cpu.amp.autocast(args...)`` is deprecated. Please use ``torch.amp.autocast("cpu", args...)`` instead.
     """
 
     def __init__(
@@ -22,6 +23,9 @@ def __init__(
             self.device = "cpu"
             self.fast_dtype = dtype
             return
+        warnings.warn(
+            "torch.cpu.amp.autocast(args...) is deprecated. Please use torch.amp.autocast('cpu', args...) instead."
+        )
         super().__init__(
             "cpu", enabled=enabled, dtype=dtype, cache_enabled=cache_enabled
         )
diff --git a/torch/cuda/amp/autocast_mode.py b/torch/cuda/amp/autocast_mode.py
@@ -1,5 +1,6 @@
 import collections
 import functools
+import warnings
 
 import torch
 
@@ -17,7 +18,7 @@
 class autocast(torch.amp.autocast_mode.autocast):
     r"""See :class:`torch.autocast`.
 
-    ``torch.cuda.amp.autocast(args...)`` is equivalent to ``torch.autocast("cuda", args...)``
+    ``torch.cuda.amp.autocast(args...)`` is deprecated. Please use ``torch.amp.autocast("cuda", args...)`` instead.
     """
 
     def __init__(
@@ -31,6 +32,9 @@ def __init__(
             self.device = "cuda"
             self.fast_dtype = dtype
             return
+        warnings.warn(
+            "torch.cuda.amp.autocast(args...) is deprecated. Please use torch.amp.autocast('cpu', args...) instead."
+        )
         super().__init__(
             "cuda", enabled=enabled, dtype=dtype, cache_enabled=cache_enabled
         )
