[Feature] support EAGLE 2 with Triton Backend

[zhyncs]

Checklist

• 1. If the issue you raised is not a feature but a question, please raise a discussion at https://github.com/sgl-project/sglang/discussions/new/choose Otherwise, it will be closed.
• 2. Please use English, otherwise it will be closed.

Motivation

Currently, EAGLE 2 is supported with FlashInfer Backend, we also need Triton Backend support. Something like custom mask support in Triton Backend needs to be implemented.

Related resources

No response

[xiefan46]

Hi @zhyncs , is there anyone working on this issue right now? I am interested in working on this issue. Do you mind sending me some code pointers for reference?

[NonvolatileMemory]

https://github.com/NonvolatileMemory/flash_tree_attn/blob/main/flash_tree_attn.py
you can see my kernel

[ispobock]

#3466