Read more about it in the official documentation of flash-attn repository.