Skip to content

sdpa: support attn_mask.requires_grad, support expanded number of heads in attn_mask #2036

sdpa: support attn_mask.requires_grad, support expanded number of heads in attn_mask

sdpa: support attn_mask.requires_grad, support expanded number of heads in attn_mask #2036

Triggered via pull request December 17, 2024 14:18
@kiya00kiya00
ready_for_review #1563
fixsdpa
Status Success
Total duration 25s
Artifacts

auto-cc.yml

on: pull_request
Fit to window
Zoom out
Zoom in

Annotations

1 warning
auto-cc
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636