Update documentation for set_flush_denormal

*set_flush_denormal() is now supported on ARM CPU's. PR:pytorch#115184 Signed-off-by: maajidkhann <maajidkhan.n@fujitsu.com>
maajidkhann · Feb 7, 2024 · 655c797 · 655c797
1 parent 0f478d9
commit 655c797
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/_torch_docs.py b/torch/_torch_docs.py
@@ -9872,7 +9872,7 @@ def merge_dicts(*dicts):
 
 Returns ``True`` if your system supports flushing denormal numbers and it
 successfully configures flush denormal mode.  :meth:`~torch.set_flush_denormal`
-is only supported on x86 architectures supporting SSE3.
+is supported on x86 architectures supporting SSE3 and ARM architecture.
 
 Args:
     mode (bool): Controls whether to enable flush denormal mode or not