How can I support other Lk?

[matianyun]

dynamic-sparse-flash-attention/openwebtext2-experiments/src/models/qksparse_attn.py

Line 264 in 266f36a

assert Lk in {64} # TODO: fix other cases

[matianyun]

Can I support other Lk such as 72?

[miraclezqc]

Can I support other Lk such as 72?

+1 Can direct annotations on this line of code be functional?

[Sakshi09Ch]

+1 is there a way to support Lk=128?