Skip to content

Commit

Permalink
Update documentation for set_flush_denormal support on ARM (#119354)
Browse files Browse the repository at this point in the history
**Documentation update for set_flush_denormal():**
-> set_flush_denormal() is now supported on ARM CPU's.
-> **PR:** #115184  (Already merged)

**Reference page:** https://pytorch.org/docs/stable/generated/torch.set_flush_denormal.html

Pull Request resolved: #119354
Approved by: https://github.com/drisspg
  • Loading branch information
maajidkhann authored and clee2000 committed Feb 14, 2024
1 parent 262047b commit aa53599
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/_torch_docs.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 AArch64 architecture.
Args:
mode (bool): Controls whether to enable flush denormal mode or not
Expand Down

0 comments on commit aa53599

Please sign in to comment.