EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Update benchmark scripts (#8)

Browse Source
This commit is contained in:
Lianmin Zheng
2024-01-15 16:12:57 -08:00
committed by GitHub
parent 01ca82d765
commit 70359bf31a
28 changed files with 183 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
docs/flashinfer.md
View File

@@ -1,20 +1,23 @@
## Flashinfer Mode
[`flashinfer`](https://github.com/flashinfer-ai/flashinfer) is a kernel library for LLM serving; we use it here to support our attention computation.
[flashinfer](https://github.com/flashinfer-ai/flashinfer) is a kernel library for LLM serving.
It can be used in SGLang runtime to accelerate attention computation.
### Install flashinfer
Note: The compilation can take a very long time.
```bash
git submodule update --init --recursive
pip install 3rdparty/flashinfer/python
```
### Run Sever With Flashinfer Mode
### Run a Server With Flashinfer Mode
Add through `--model_mode` argument from the command line.
Add `--model-mode flashinfer` argument to enable flashinfer when launching a server.
Example:
```bash
python -m sglang.launch_server --model-path meta-llama/Llama-2-7b-chat-hf --port 30000 --model-mode flashinfer
```
```