sdpa: support attn_mask.requires_grad, support expanded number of heads in attn_mask #4553
kiya00Annotations
1 warning
|
triage-prs
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636
|