sdpa: support attn_mask.requires_grad, support expanded number of heads in attn_mask · Lightning-AI/lightning-thunder@e9d525e

Triggered via pull request December 17, 2024 14:18

kiya00

ready_for_review #1563

fixsdpa

Status Success

Total duration 3m 26s

Artifacts 1

docs-build.yml

on: pull_request

Matrix: docs-make

1 error

Process completed with exit code 2.

Produced during runtime

Name	Size
docs-html-ac3b5801dc6049090338653c536c8814369912d6	4.47 MB