pytorch
51c6c5e1 - [SDPA] Standardizes the return shape for dense tensor of SDPA regardless of fused kernel called (#90776)

Commit
2 years ago
[SDPA] Standardizes the return shape for dense tensor of SDPA regardless of fused kernel called (#90776) # Summary Continues to fix up the meta output story of SDPA to be more correct Pull Request resolved: https://github.com/pytorch/pytorch/pull/90776 Approved by: https://github.com/cpuhrsch
Author
Committer
Parents
Loading