Upgrade fa3 hash (#28394)

Co-authored-by: Shijin <dovis.zhang02@gmail.com>
2026-06-17 13:32:11 -07:00
parent 753aa89a83
commit 4b817f5d7f
6 changed files with 82 additions and 16 deletions
@@ -82,7 +82,9 @@ std::tuple<at::Tensor, at::Tensor, at::Tensor, at::Tensor> mha_fwd(
    int64_t num_splits,
    std::optional<bool> pack_gqa_,
    int64_t sm_margin,
-    std::optional<const at::Tensor>& sinks_);  // (h)
+    std::optional<const at::Tensor>& sinks_,      // (h)
+    std::optional<at::Tensor> sparse_mask_fine_,  // [total_q, max_k_blocks, num_int32_per_block]
+    bool only_qv);

 /*
 * From flash-attention: get_scheduler_metadata