from torch._torch_docs import _add_docstr

_NPU_SDPA_NOTE = r"""

Note (NPU):

On NPU, ``scaled_dot_product_attention`` dispatches to CANN ``aclnnFusionAttention``
family kernels. ...

"""

_add_docstr(
torch.nn.functional.scaled_dot_product_attention,
(torch.nn.functional.scaled_dot_product_attention.doc or "") + _NPU_SDPA_NOTE,
)