pytorch
7265f60a - Regularize mask handling for attn_mask and key_padding_mask (#92733)

Commit
2 years ago
Regularize mask handling for attn_mask and key_padding_mask (#92733) Summary: Regularize mask handling for attn_mask and key_padding_mask * Update documentation to remove reference to byte masks (which were deprecated long ago) * Introduce check and warn about deprecation if attn_mask and key_padding_mask types mismatch * Convert all masks to float before combining * Combine by adding Test Plan: sandcastle & github CI Differential Revision: D42653215 Pull Request resolved: https://github.com/pytorch/pytorch/pull/92733 Approved by: https://github.com/ngimel, https://github.com/drisspg
Author
Michael Gschwind
Committer
Parents
Loading