InternLM / InternLM/lmdeploy

[Bug] PyTorch Engine poor performance compared to vllm

Open
#1,449 3 comments 0 reactions 1 assignee Claimed by @grimoire View on GitHub
Dominant language
Python
Stars
8.1k
Forks
748
Avg merge
6d 2h
Merged PRs (30d)
54

Description

### Checklist

- [x] 1. I have searched related issues but cannot get the expected help.
- [x] 2. The bug has not been fixed in the latest version.

### Describe the bug

I tried to benchmark the PyTorch Engine performance and find it very poor...

`PyTorch Engine`
concurrency: 4
input token Throughput: 101.53 tokens/s
output token Throughput: 93.32 tokens/s
total token Throughput: 194.85 tokens/s

`vllm`
concurrency: 4
input token Throughput: 184.18 tokens/s
output token Throughput: 169.28 tokens/s
total token Throughput: 353.46 tokens/s

Is it normal? Do I miss something when use PyTorch Engine?

### Reproduction

model: Qwen14B
GPU: A100
LMDeploy: 0.3.0
Dataset: ShareGPT_V3_unfiltered_cleaned_split.json
profile_restful_api.py

### Environment

```Shell
LMDeploy: 0.3.0
```

### Error traceback

_No response_

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.