[CUDA] Fix MHA mask (#21655)

Commit

1 year ago

[CUDA] Fix MHA mask (#21655) ### Description Fix a check of mask type introduced by me in a recent commit. Add tests.