pytorch
diff --git a/‎torch/cpu/amp/autocast_mode.py
Copy file name to clipboardExpand all lines: torch/cpu/amp/autocast_mode.py
+5-1Lines changed: 5 additions & 1 deletion b/‎torch/cpu/amp/autocast_mode.py
Copy file name to clipboardExpand all lines: torch/cpu/amp/autocast_mode.py
+5-1Lines changed: 5 additions & 1 deletion
diff --git a/‎torch/cuda/amp/autocast_mode.py
Copy file name to clipboardExpand all lines: torch/cuda/amp/autocast_mode.py
+5-1Lines changed: 5 additions & 1 deletion b/‎torch/cuda/amp/autocast_mode.py
Copy file name to clipboardExpand all lines: torch/cuda/amp/autocast_mode.py
+5-1Lines changed: 5 additions & 1 deletion
@@ -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
         )
 
@@ -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
         )