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

Need parallel linears #50

Open
3 tasks
EricLBuehler opened this issue Apr 1, 2024 · 4 comments
Open
3 tasks

Need parallel linears #50

EricLBuehler opened this issue Apr 1, 2024 · 4 comments
Labels
backend Backend work paged-attention

Comments

@EricLBuehler
Copy link
Owner

EricLBuehler commented Apr 1, 2024 •
edited

  • RowParallelLinear
  • MergedColumnParallelLinear
  • QKVParallelLinear
@hugoabonizio
Copy link
Contributor

Is the plan to implement tensor sharding for both quantized and non-quantized versions?

@EricLBuehler EricLBuehler added the backend Backend work label Apr 3, 2024
@EricLBuehler
Copy link
Owner Author

Yes, that is the plan.

@hugoabonizio
Copy link
Contributor

I'm quite interested in the parallelization in CUDA of the quantized models!

@EricLBuehler
Copy link
Owner Author

Yes! We are beginning work on this topic now.

@EricLBuehler EricLBuehler mentioned this issue Apr 4, 2024
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
backend Backend work paged-attention
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@hugoabonizio @EricLBuehler