Skip to content

Fix double transpose of K/V in templated attention backward - #14341

Open
guptaishaan wants to merge 2 commits into
huggingface:mainfrom
guptaishaan:fix-14338
Open

Fix double transpose of K/V in templated attention backward#14341
guptaishaan wants to merge 2 commits into
huggingface:mainfrom
guptaishaan:fix-14338

Cover the num_heads != seq_len case in the attention op test

0a1f670
Select commit
Loading
Failed to load commit list.
Sign in for the full log view