-
Notifications
You must be signed in to change notification settings - Fork 234
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
Support mixtral kvcache reuse and remove kv_cache_fp8 #898
Support mixtral kvcache reuse and remove kv_cache_fp8 #898
Conversation
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
Hi @regisss, could you help review this PR, it's based on your PR 'linear and dynamic rope', thanks! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!!!
@jychen-habana nice PR and we could enable |
Hi @regisss could you please review and merge this PR? |
Co-authored-by: regisss <15324346+regisss@users.noreply.github.com>
What does this PR do?
Before
input 32: ('Peace is the only way',)
output 1: ('Peace is the only way to end the war in Syria.\n\nThe Syrian people have suffered enough.\n\nThe war has killed hundreds of thousands of people, displaced millions, and destroyed the country’s infrastructure.\n\nThe international community must do everything in its power to end the conflict and bring peace to Syria.\n\nThe United Nations Security Council should pass a resolution calling for an immediate ceasefire and the start of negotiations between the Syrian government and the opposition.\n\nThe international community should also provide humanitarian assistance to the Syrian people and help rebuild the country.\n\nOnly by working together can we hope to achieve peace in Syria.\n\n## What is the best way to end the war in Syria?\n\nThe best way to end the war in Syria is to find a political solution that includes all parties involved. This would require a ceasefire and negotiations between the Syrian government and the opposition. The international community should also provide humanitarian assistance to the Syrian people and help rebuild the country.\n\n## How can the international community help to end the war in Syria?\n\nThe international community can help to end the war in Syria by providing humanitarian assistance to the Syrian people and helping to rebuild the country.',)
Stats:
Throughput (including tokenization) = 1136.5631648005235 tokens/second
Number of HPU graphs = 183
Memory allocated = 58.67 GB
Max memory allocated = 60.26 GB
Total memory available = 94.62 GB
Graph compilation duration = 227.78881454397924 seconds
After
input 32: ('Peace is the only way',)
output 1: ('Peace is the only way to end the war in Syria.\n\nThe Syrian people have suffered enough.\n\nThe war has killed hundreds of thousands of people, displaced millions, and destroyed the country’s infrastructure.\n\nThe international community must do everything in its power to end the conflict and bring peace to Syria.\n\nThe United Nations Security Council should pass a resolution calling for an immediate ceasefire and the start of negotiations between the Syrian government and the opposition.\n\nThe international community should also provide humanitarian assistance to the Syrian people and help rebuild the country.\n\nOnly by working together can we hope to achieve peace in Syria.\n\n## What is the best way to end the war in Syria?\n\nThe best way to end the war in Syria is to find a political solution that includes all parties involved. This would require a ceasefire and negotiations between the Syrian government and the opposition. The international community should also provide humanitarian assistance to the Syrian people and help rebuild the country.\n\n## How can the international community help to end the war in Syria?\n\nThe international community can help to end the war in Syria by providing humanitarian assistance to the Syrian people and helping to rebuild the country.',)
Stats:
Throughput (including tokenization) = 1221.6872883013737 tokens/second
Number of HPU graphs = 180
Memory allocated = 52.36 GB
Max memory allocated = 52.45 GB
Total memory available = 94.62 GB
Graph compilation duration = 216.56420183897717 seconds
Before submitting