Passing nonexistent argument when flash_attn version is >= 2.5.7

[MaciejBalaNV]

In this place in TE we pass "block_table" argument to either flash_attn_func or flash_attn_varlen_func functions from flash_attn package. However, this function doesn't have this argument, only flash_attn_varlen_func function does, as we can verify here. I'm attaching HEAD of main as a link, but it's the same in version 2.5.8. It results in an error when flash_attn_func is chosen.

[MaciejBalaNV]

Seems like this bug was added in this commit, which introduced support for FA3.