How to use the function of flash-attn-1 to mimic the behavior of flash_attn_func in flash-attn-2?

[jpWang]

Hi,
thanks for this great repo~
And I want to ask how can I use the functions in flash-attn-v1 to mimic the behavior of the flash_attn_func function in flash-attn-v2 ?
Due to the environment I am using, I can only use flash-attn-v1. Can I use the same input to obtain the same output? And I don't care about the speed.
Thank you in advance for your assistance.