Improve use of torch.is_autocast_enabled (#43930)
* Improve use of torch.is_autocast_enabled
Signed-off-by: Yuanyuan Chen <cyyever@outlook.com>
* Fix documentation examples
Signed-off-by: Yuanyuan Chen <cyyever@outlook.com>
---------
Signed-off-by: Yuanyuan Chen <cyyever@outlook.com>