pytorch
0db704d2 - [OpInfo] Add multi_head_attention_forward (#100153)

Commit
1 year ago
[OpInfo] Add multi_head_attention_forward (#100153) <!-- copilot:summary --> ### <samp>🤖 Generated by Copilot at 8f8d620</samp> This pull request improves the testing of the `nn.functional.multi_head_attention_forward` function by adding it to the `OpInfo` framework, adjusting the tolerance and skipping criteria for some test cases, and restricting the dtype for the `MetaProgrammingSystem` tests. These changes aim to address the randomness and numerical precision issues of the function. Pull Request resolved: https://github.com/pytorch/pytorch/pull/100153 Approved by: https://github.com/drisspg
Author
Committer
Parents
Loading