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

[Usage]: The TP improvement is not as expectation #274

Open
JunxiChhen opened this issue Sep 12, 2024 · 1 comment
Open

[Usage]: The TP improvement is not as expectation #274

JunxiChhen opened this issue Sep 12, 2024 · 1 comment

Comments

@JunxiChhen
Copy link

Your current environment

The offline inference of Llama-3-8B with benchmark_latency.py sweeping on 1, 2, 4 cards results:
image

And the optimum-habana results:
image

The results show that on 1 card vLLM is greater than optimum-habana. But when inference on multi-card, the TP in vLLM performance gain is not good enough, so that the performance is worse than optimum-habana.

How would you like to use vllm

I want to run inference of a [specific model](put link here). I don't know how to integrate it with vllm.

@wpyszka
Copy link

wpyszka commented Sep 24, 2024

@JunxiChhen, Vllm performance improvements are planned in SW1.19 version. Please stay tuned.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants