Add treemask mode to build_eagle_tree & release sgl-kernel 0.2.3 (#7756)

Co-authored-by: Pranjal Shankhdhar <pranjal.ssh@gmail.com>
2025-07-05 12:17:05 -07:00
parent c04a8a820b
commit 5589b75024
6 changed files with 101 additions and 36 deletions
--- a/sgl-kernel/include/sgl_kernel_ops.h
+++ b/sgl-kernel/include/sgl_kernel_ops.h
@@ -374,7 +374,8 @@ void build_tree_kernel_efficient(
    at::Tensor retrive_next_sibling,
    int64_t topk,
    int64_t depth,
-    int64_t draft_token_num);
+    int64_t draft_token_num,
+    int64_t tree_mask_mode);

 void segment_packbits(
    at::Tensor x,