That's Flash Attention 1, but yeah there's nothing stopping someone from submitting a PR.

This enables V1. V2 is still yet to be integrated into xformers. The team replied saying it should happen this week.

See the relevant Github issue here: https://github.com/facebookresearch/xformers/issues/795