frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed typo error (#159451)

Browse Source 
Fixes #159375

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159451
Approved by: https://github.com/albanD
This commit is contained in:
Rohit Manav
2025-07-30 17:41:26 +00:00
committed by PyTorch MergeBot
parent 1b99c1859c
commit 5c14315b05
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/backends/cuda/__init__.py
View File

@ -294,7 +294,7 @@ def preferred_rocm_fa_library(
[ROCm-only]
Override the backend PyTorch uses in ROCm environments for Flash Attention. Choose between AOTriton and CK
.. warning:: This flag is experimeental and subject to change.
.. warning:: This flag is experimental and subject to change.
When Flash Attention is enabled and desired, PyTorch defaults to using AOTriton as the backend.
This flag (a :class:`str`) allows users to override this backend to use composable_kernel