Feature(MInference): fix the func name

minference/ops/block_sparse_flash_attention.py

@@ -444,7 +444,7 @@ def test_flash_attention(
     print('========================================\n')
 
 
-def block_sparse_flash_attention_forward(
+def block_sparse_attention(
     query: torch.Tensor,  # [BATCH, N_HEADS, N_CTX, D_HEAD]
     key: torch.Tensor,    # [BATCH, N_HEADS, N_CTX, D_HEAD]
     value: torch.Tensor,  # [BATCH, N_HEADS, N_CTX, D_HEAD]

minference/ops/pit_sparse_flash_attention_v2.py

@@ -693,7 +693,7 @@ def test_flash_attention(
         torch.testing.assert_close(output_flash, output_triton_sparse, atol=1e-2, rtol=0)
 
 
-def pit_sparse_flash_attention_forward(
+def vertical_slash_sparse_attention(
     query: torch.Tensor,  # [BATCH, N_HEADS, N_CTX, D_HEAD]
     key: torch.Tensor,    # [BATCH, N_HEADS, N_CTX, D_HEAD]
     value: torch.Tensor,  # [BATCH, N_HEADS, N_CTX, D_HEAD]