Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Eagle speculative decoding part 1: Support target model verification in the attention backend #2678

Merged
merged 1 commit into from
Dec 31, 2024

Conversation

merrymercy
Copy link
Contributor

@merrymercy merrymercy commented Dec 31, 2024

This is used for speculative decoding in #2150

Co-authored-by: yukavio <[email protected]>

@merrymercy merrymercy changed the title Support target verify in the attention backend Support target model verification in the attention backend Dec 31, 2024
@merrymercy merrymercy merged commit f44d143 into main Dec 31, 2024
16 of 17 checks passed
@merrymercy merrymercy deleted the pr-attn branch December 31, 2024 06:58
@zhyncs
Copy link
Member

zhyncs commented Dec 31, 2024

@merrymercy test_dp_attention failed after this PR

python3 test_dp_attention.py

@zhyncs
Copy link
Member

zhyncs commented Dec 31, 2024

FYI After I revert git revert f44d143949f6c6fbca6cb96c52381b8bc1769a87, it works well cc @ispobock

@merrymercy merrymercy changed the title Support target model verification in the attention backend Eagle speculative decoding part 1. Support target model verification in the attention backend Dec 31, 2024
@merrymercy merrymercy changed the title Eagle speculative decoding part 1. Support target model verification in the attention backend Eagle speculative decoding part 1: Support target model verification in the attention backend Dec 31, 2024
@merrymercy
Copy link
Contributor Author

The hanging issue is fixed by #2684

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants