Update on "[doc] fix amp doc"

# Motivation The doc shown on https://pytorch.org/docs/main/amp.html#torch.amp.autocast_mode.is_autocast_available should be `torch.amp.is_autocast_available` instead of `torch.amp.autocast_mode.is_autocast_available`. This PR aims to fix it. cc mcarilli ptrblck leslie-fang-intel jgong5 [ghstack-poisoned]
pytorch · May 11, 2024 · 0865d7e · 0865d7e
2 parents 2a6ac3b + 9ebe022
commit 0865d7e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/source/amp.rst b/docs/source/amp.rst
@@ -48,6 +48,7 @@ Autocasting
 
 .. autoclass:: DOC_TEST
     :members:
+    .. automethod:: test
 
 .. currentmodule:: torch