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.

Sign up for GitHub

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

Jump to bottom

TGI Improvements for ROCM 6.2 #151

Open
7 of 8 tasks
mht-sharma opened this issue Sep 11, 2024 · 1 comment
Open
7 of 8 tasks

TGI Improvements for ROCM 6.2 #151

mht-sharma opened this issue Sep 11, 2024 · 1 comment
Assignees
@mht-sharma
Labels
Instinct

Comments

@mht-sharma
Copy link
Collaborator

mht-sharma commented Sep 11, 2024 •
edited
Loading

  • Update CK flash attention to use CK tile
  • Update VLLM to latest rocm/vllm commit.
  • Update torch
  • Fix tunable op issue with TP8
  • BF16 inference fix
  • Custom Paged attention
  • Documentation
  • Rebase
@mht-sharma
Copy link
Collaborator Author

PR ready: huggingface/text-generation-inference#2358

@mht-sharma mht-sharma self-assigned this Sep 11, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@mht-sharma mht-sharma

Labels
Instinct
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@mht-sharma