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

Fix turbomind TP #2706

Merged
merged 2 commits into from
Nov 5, 2024
Merged

Fix turbomind TP #2706

merged 2 commits into from
Nov 5, 2024

Conversation

lzhangzz
Copy link
Collaborator

@lzhangzz lzhangzz commented Nov 4, 2024

Wait for pinned buffers to be allocated for all ranks, otherwise tuning will hang due to concurrent kernel launch & cudaMallocHost.

@lvhan028 lvhan028 changed the base branch from main to 0.6.2.post1 November 4, 2024 11:54
@lvhan028 lvhan028 changed the base branch from 0.6.2.post1 to main November 4, 2024 11:54
@lvhan028 lvhan028 mentioned this pull request Nov 5, 2024
2 tasks
@lvhan028 lvhan028 merged commit 71f1d0f into InternLM:main Nov 5, 2024
9 checks passed
AllentDan pushed a commit to AllentDan/lmdeploy that referenced this pull request Nov 13, 2024
* fix tp

* use fused moe for all arches by default
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants