README.md 11.9 KB
Newer Older
1
# Contributing to vLLM
2
3
4
5
6
7
8
9
10
11
12
13

Thank you for your interest in contributing to vLLM! Our community is open to everyone and welcomes all kinds of contributions, no matter how small or large. There are several ways you can contribute to the project:

- Identify and report any issues or bugs.
- Request or add support for a new model.
- Suggest or implement new features.
- Improve documentation or contribute a how-to guide.

We also believe in the power of community support; thus, answering queries, offering PR reviews, and assisting others are also highly regarded and beneficial contributions.

Finally, one of the most impactful ways to support us is by raising awareness about vLLM. Talk about it in your blog posts and highlight how it's driving your incredible projects. Express your support on social media if you're using vLLM, or simply offer your appreciation by starring our repository!

14
15
16
17
18
## Job Board

Unsure on where to start? Check out the following links for tasks to work on:

- [Good first issues](https://github.com/vllm-project/vllm/issues?q=is%3Aissue%20state%3Aopen%20label%3A%22good%20first%20issue%22)
19
    - [Selected onboarding tasks](https://github.com/orgs/vllm-project/projects/6)
20
- [New model requests](https://github.com/vllm-project/vllm/issues?q=is%3Aissue%20state%3Aopen%20label%3A%22new-model%22)
21
    - [Models with multi-modal capabilities](https://github.com/orgs/vllm-project/projects/10)
22

23
## License
24

25
See [LICENSE](../../LICENSE).
26

27
## Developing
28

29
The first step of contributing to vLLM is to clone the GitHub repository:
30

31
32
33
34
```bash
git clone https://github.com/vllm-project/vllm.git
cd vllm
```
35

36
Then, configure your Python virtual environment.
37

38
--8<-- "docs/getting_started/installation/python_env_setup.inc.md"
Reid's avatar
Reid committed
39

40
If you are only developing vLLM's Python code, install vLLM using:
Reid's avatar
Reid committed
41

42
43
44
```bash
VLLM_USE_PRECOMPILED=1 uv pip install -e .
```
Reid's avatar
Reid committed
45

46
If you are developing vLLM's Python and CUDA/C++ code, install Pytorch first:
47
48

```bash
49
50
51
uv pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu129
```

52
53
54
55
56
57
58
Then install the necessary build dependencies from `requirements/build.txt`, skipping `torch` as it was installed in the previous step:

```bash
grep -v '^torch==' requirements/build.txt | uv pip install -r -
```

Finally install vLLM using:
59
60
61

```bash
uv pip install -e . --no-build-isolation
62
63
```

64
For more details about installing from source and installing for other hardware, check out the [installation instructions](../getting_started/installation/README.md) for your hardware and head to the "Build wheel from source" section.
Reid's avatar
Reid committed
65

66
For an optimized workflow when iterating on C++/CUDA kernels, see the [Incremental Compilation Workflow](./incremental_build.md) for recommendations.
Reid's avatar
Reid committed
67

68
!!! tip
69
    vLLM is compatible with Python versions 3.10 to 3.13. However, vLLM's default [Dockerfile](../../docker/Dockerfile) ships with Python 3.12 and tests in CI (except `mypy`) are run with Python 3.12.
Reid's avatar
Reid committed
70

71
    Therefore, we recommend developing with Python 3.12 to minimise the chance of your local environment clashing with our CI environment.
Reid's avatar
Reid committed
72

73
### Linting
Reid's avatar
Reid committed
74

75
vLLM uses `pre-commit` to lint and format the codebase. See <https://pre-commit.com/#usage> if `pre-commit` is new to you. Setting up `pre-commit` is as easy as:
Reid's avatar
Reid committed
76
77

```bash
78
79
uv pip install pre-commit
pre-commit install
Reid's avatar
Reid committed
80
81
```

82
vLLM's `pre-commit` hooks will now run automatically every time you commit.
Reid's avatar
Reid committed
83

84
85
!!! tip "Tips"
    You can manually run the `pre-commit` hooks using:
86

87
88
89
90
    ```bash
    pre-commit run     # runs on staged files
    pre-commit run -a  # runs on all files (short for --all-files)
    ```
91

92
    ---
Reid's avatar
Reid committed
93

94
    Some `pre-commit` hooks only run in CI. If you need to, you can run them locally with:
Reid's avatar
Reid committed
95

96
97
    ```bash
    pre-commit run --hook-stage manual markdownlint
98
    pre-commit run --hook-stage manual mypy-3.10
99
    ```
Reid's avatar
Reid committed
100

101
### Documentation
Reid's avatar
Reid committed
102

103
MkDocs is a fast, simple and downright gorgeous static site generator that's geared towards building project documentation. Documentation source files are written in Markdown, and configured with a single YAML configuration file, [mkdocs.yaml](../../mkdocs.yaml).
Reid's avatar
Reid committed
104

105
Get started with:
Reid's avatar
Reid committed
106

107
108
109
```bash
uv pip install -r requirements/docs.txt
```
110

111
112
113
!!! tip
    Ensure that your Python version is compatible with the plugins
    (e.g., `mkdocs-awesome-nav` requires Python 3.10+)
114

115
116
MkDocs comes with a built-in dev-server that lets you preview your documentation as you work on it.
From the root of the repository, run:
117

118
119
120
121
```bash
mkdocs serve                           # with API ref (~10 minutes)
API_AUTONAV_EXCLUDE=vllm mkdocs serve  # API ref off (~15 seconds)
```
122

123
124
Once you see `Serving on http://127.0.0.1:8000/` in the logs, the live preview is ready!
Open <http://127.0.0.1:8000/> in your browser to see it.
125

126
For additional features and advanced configurations, refer to the:
Reid's avatar
Reid committed
127

128
129
- [MkDocs documentation](https://www.mkdocs.org/)
- [Material for MkDocs documentation](https://squidfunk.github.io/mkdocs-material/) (the MkDocs theme we use)
130

131
### Testing
132

133
vLLM uses `pytest` to test the codebase.
134

135
136
137
138
139
140
141
142
143
```bash
# Install the test dependencies used in CI (CUDA only)
uv pip install -r requirements/common.txt -r requirements/dev.txt --torch-backend=auto

# Install some common test dependencies (hardware agnostic)
uv pip install pytest pytest-asyncio

# Run all tests
pytest tests/
144

145
146
147
148
149
# Run tests for a single test file with detailed output
pytest -s -v tests/test_logger.py
```

!!! tip "Install python3-dev if Python.h is missing"
150
151
152
    If any of the above commands fails with `Python.h: No such file or directory`, install
    `python3-dev` with `sudo apt install python3-dev`.

153
!!! warning "Warnings"
154
    Currently, the repository is not fully checked by `mypy`.
155

156
157
    ---

158
159
160
    Currently, not all unit tests pass when run on CPU platforms. If you don't have access to a GPU
    platform to run unit tests locally, rely on the continuous integration system to run the tests for
    now.
161

162
## Issues
163

164
If you encounter a bug or have a feature request, please [search existing issues](https://github.com/vllm-project/vllm/issues?q=is%3Aissue) first to see if it has already been reported. If not, please [file a new issue](https://github.com/vllm-project/vllm/issues/new/choose), providing as much relevant information as possible.
165

166
!!! important
167
    If you discover a security vulnerability, please follow the instructions [here](../../SECURITY.md).
168

169
## Pull Requests & Code Reviews
170

171
172
173
174
Thank you for your contribution to vLLM! Before submitting the pull request,
please ensure the PR meets the following criteria. This helps vLLM maintain the
code quality and improve the efficiency of the review process.

175
### DCO and Signed-off-by
176

177
When contributing changes to this project, you must agree to the [DCO](../../DCO).
178
Commits must include a `Signed-off-by:` header which certifies agreement with
179
the terms of the DCO.
180

181
Using `-s` with `git commit` will automatically add this header.
182

183
!!! tip
184
    You can enable automatic sign-off via your IDE:
185

186
187
188
189
    - **PyCharm**: Click on the `Show Commit Options` icon to the right of the `Commit and Push...` button in the `Commit` window.
      It will bring up a `git` window where you can modify the `Author` and enable `Sign-off commit`.
    - **VSCode**: Open the [Settings editor](https://code.visualstudio.com/docs/configure/settings)
      and enable the `Git: Always Sign Off` (`git.alwaysSignOff`) field.
190

191
### PR Title and Classification
192
193
194
195

Only specific types of PRs will be reviewed. The PR title is prefixed
appropriately to indicate the type of change. Please use one of the following:

196
197
198
199
- `[Bugfix]` for bug fixes.
- `[CI/Build]` for build or continuous integration improvements.
- `[Doc]` for documentation fixes and improvements.
- `[Model]` for adding a new model or improving an existing model. Model name
200
  should appear in the title.
201
202
203
204
205
206
207
208
- `[Frontend]` For changes on the vLLM frontend (e.g., OpenAI API server,
  `LLM` class, etc.)
- `[Kernel]` for changes affecting CUDA kernels or other compute kernels.
- `[Core]` for changes in the core vLLM logic (e.g., `LLMEngine`,
  `AsyncLLMEngine`, `Scheduler`, etc.)
- `[Hardware][Vendor]` for hardware-specific changes. Vendor name should
  appear in the prefix (e.g., `[Hardware][AMD]`).
- `[Misc]` for PRs that do not fit the above categories. Please use this
209
210
  sparingly.

211
212
!!! note
    If the PR spans more than one category, please include all relevant prefixes.
213

214
### Code Quality
215
216
217

The PR needs to meet the following code quality standards:

218
- We adhere to [Google Python style guide](https://google.github.io/styleguide/pyguide.html) and [Google C++ style guide](https://google.github.io/styleguide/cppguide.html).
219
- Pass all linter checks.
220
221
222
223
- The code needs to be well-documented to ensure future contributors can easily
  understand the code.
- Include sufficient tests to ensure the project stays correct and robust. This
  includes both unit tests and integration tests.
224
225
- Please add documentation to `docs/` if the PR modifies the user-facing behaviors of vLLM.
  It helps vLLM users understand and utilize the new features or changes.
226

227
### Adding or Changing Kernels
228

229
When actively developing or modifying kernels, using the [Incremental Compilation Workflow](./incremental_build.md) is highly recommended for faster build times.
230
231
232
Each custom kernel needs a schema and one or more implementations to be registered with PyTorch.

- Make sure custom ops are registered following PyTorch guidelines:
233
234
235
  [Custom C++ and CUDA Operators](https://pytorch.org/tutorials/advanced/cpp_custom_ops.html#cpp-custom-ops-tutorial)
  and [The Custom Operators Manual](https://docs.google.com/document/d/1_W62p8WJOQQUzPsJYa7s701JXt0qf2OfLub2sbkHOaU).
- Custom operations that return `Tensors` require meta-functions.
236
237
238
  Meta-functions should be implemented and registered in Python so that dynamic
  dims can be handled automatically. See above documents for a description of
  meta-functions.
239
- Use [torch.library.opcheck()](https://pytorch.org/docs/stable/library.html#torch.library.opcheck)
240
  to test the function registration and meta-function for any registered ops.
241
  See `tests/kernels` for examples.
242
243
244
- When changing the C++ signature of an existing op, the schema must be updated
  to reflect the changes.
- If a new custom type is needed, see the following document:
245
  [Custom Class Support in PT2](https://docs.google.com/document/d/18fBMPuOJ0fY5ZQ6YyrHUppw9FA332CpNtgB6SOIgyuA).
246

247
### Notes for Large Changes
248
249
250
251

Please keep the changes as concise as possible. For major architectural changes
(>500 LOC excluding kernel/data/config/test), we would expect a GitHub issue
(RFC) discussing the technical design and justification. Otherwise, we will tag
252
it with `rfc-required` and might not go through the PR.
253

254
### What to Expect for the Reviews
255
256
257
258
259
260
261
262
263
264
265
266

The goal of the vLLM team is to be a *transparent reviewing machine*. We would
like to make the review process transparent and efficient and make sure no
contributor feels confused or frustrated. However, the vLLM team is small, so we
need to prioritize some PRs over others. Here is what you can expect from the
review process:

- After the PR is submitted, the PR will be assigned to a reviewer. Every
  reviewer will pick up the PRs based on their expertise and availability.
- After the PR is assigned, the reviewer will provide status updates every 2-3
  days. If the PR is not reviewed within 7 days, please feel free to ping the
  reviewer or the vLLM team.
267
- After the review, the reviewer will put an `action-required` label on the PR
268
269
270
271
272
  if there are changes required. The contributor should address the comments and
  ping the reviewer to re-review the PR.
- Please respond to all comments within a reasonable time frame. If a comment
  isn't clear or you disagree with a suggestion, feel free to ask for
  clarification or discuss the suggestion.
273
274
275
- Note that not all CI checks will be executed due to limited computational
  resources. The reviewer will add `ready` label to the PR when the PR is
  ready to merge or a full CI run is needed.
276

277
## Thank You
278
279
280

Finally, thank you for taking the time to read these guidelines and for your interest in contributing to vLLM.
All of your contributions help make vLLM a great tool and community for everyone!