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

[Distribute] Add distribute support for continuous batching api. #421

Merged
merged 3 commits into from
Jun 4, 2024

Conversation

Duyi-Wang
Copy link
Contributor

Master and slaves should both run according to the following workflow:

while True:
    model.set_input_cb()
    model.forward_cb()
    model.free_seqs()

@Duyi-Wang Duyi-Wang added enhancement New feature or request continuous batching continuous batching labels May 24, 2024
@Duyi-Wang Duyi-Wang requested a review from pujiang2018 May 24, 2024 08:34
@Duyi-Wang Duyi-Wang marked this pull request as draft May 24, 2024 08:51
@Duyi-Wang
Copy link
Contributor Author

Need to fix following Error after complete all requests in vllm:

Abort(68817167) on node 1 (rank 1 in comm 0): Fatal error in PMPI_Test: Other MPI error, error stack:
PMPI_Test(186)..................: MPI_Test(request=0x74a0f28, flag=0x7f3d67ffdd3c, status=0x1) failed
MPIR_Test(79)...................:
MPIDU_Sched_progress_state(1060): Invalid communicator

@Duyi-Wang Duyi-Wang marked this pull request as ready for review May 28, 2024 06:09
@Duyi-Wang
Copy link
Contributor Author

@pujiang2018 Ready.

@pujiang2018 pujiang2018 merged commit d47535e into intel:main Jun 4, 2024
1 check passed
@Duyi-Wang Duyi-Wang deleted the cb_distribute branch June 5, 2024 04:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
continuous batching continuous batching enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants