sparse attention torch实现

[andy-upup]

请问你们开源的代码中，在哪里能找到sparse attention的torch实现吗

[suhmily10]

infllmv2 算子库有和 torch 的对拍单测，可以用作参考：
https://github.com/OpenBMB/infllmv2_cuda_impl/blob/main/tests/test_stage1.py
https://github.com/OpenBMB/infllmv2_cuda_impl/blob/main/tests/test_pooling_varlen.py
https://github.com/OpenBMB/infllmv2_cuda_impl/blob/main/tests/test_infllmv2.py