README.md 11.5 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](gh-project: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](gh-project:10)
22

23
## License
24

25
See <gh-file: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 vLLM using:
47
48

```bash
49
uv pip install -e .
50
51
```

52
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
53

54
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
55

56
!!! tip
57
    vLLM is compatible with Python versions 3.10 to 3.13. However, vLLM's default [Dockerfile](gh-file:docker/Dockerfile) ships with Python 3.12 and tests in CI (except `mypy`) are run with Python 3.12.
Reid's avatar
Reid committed
58

59
    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
60

61
### Linting
Reid's avatar
Reid committed
62

63
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
64
65

```bash
66
67
uv pip install pre-commit
pre-commit install
Reid's avatar
Reid committed
68
69
```

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

72
73
!!! tip "Tips"
    You can manually run the `pre-commit` hooks using:
74

75
76
77
78
    ```bash
    pre-commit run     # runs on staged files
    pre-commit run -a  # runs on all files (short for --all-files)
    ```
79

80
    ---
Reid's avatar
Reid committed
81

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

84
85
    ```bash
    pre-commit run --hook-stage manual markdownlint
86
    pre-commit run --hook-stage manual mypy-3.10
87
    ```
Reid's avatar
Reid committed
88

89
### Documentation
Reid's avatar
Reid committed
90

91
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, <gh-file:mkdocs.yaml>.
Reid's avatar
Reid committed
92

93
Get started with:
Reid's avatar
Reid committed
94

95
96
97
```bash
uv pip install -r requirements/docs.txt
```
98

99
100
101
!!! tip
    Ensure that your Python version is compatible with the plugins
    (e.g., `mkdocs-awesome-nav` requires Python 3.10+)
102

103
104
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:
105

106
107
108
109
```bash
mkdocs serve                           # with API ref (~10 minutes)
API_AUTONAV_EXCLUDE=vllm mkdocs serve  # API ref off (~15 seconds)
```
110

111
112
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.
113

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

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

119
### Testing
120

121
vLLM uses `pytest` to test the codebase.
122

123
124
125
126
127
128
129
130
131
```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/
132

133
134
135
136
137
# 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"
138
139
140
    If any of the above commands fails with `Python.h: No such file or directory`, install
    `python3-dev` with `sudo apt install python3-dev`.

141
!!! warning "Warnings"
142
    Currently, the repository is not fully checked by `mypy`.
143

144
145
    ---

146
147
148
    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.
149

150
## Issues
151

152
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.
153

154
!!! important
155
    If you discover a security vulnerability, please follow the instructions [here](gh-file:SECURITY.md#reporting-a-vulnerability).
156

157
## Pull Requests & Code Reviews
158

159
160
161
162
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.

163
### DCO and Signed-off-by
164

165
When contributing changes to this project, you must agree to the <gh-file:DCO>.
166
Commits must include a `Signed-off-by:` header which certifies agreement with
167
the terms of the DCO.
168

169
Using `-s` with `git commit` will automatically add this header.
170

171
!!! tip
172
    You can enable automatic sign-off via your IDE:
173

174
175
176
177
    - **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.
178

179
### PR Title and Classification
180
181
182
183

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:

184
185
186
187
- `[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
188
  should appear in the title.
189
190
191
192
193
194
195
196
- `[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
197
198
  sparingly.

199
200
!!! note
    If the PR spans more than one category, please include all relevant prefixes.
201

202
### Code Quality
203
204
205

The PR needs to meet the following code quality standards:

206
- 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).
207
- Pass all linter checks.
208
209
210
211
- 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.
212
213
- 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.
214

215
### Adding or Changing Kernels
216

217
When actively developing or modifying kernels, using the [Incremental Compilation Workflow](./incremental_build.md) is highly recommended for faster build times.
218
219
220
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:
221
222
223
  [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.
224
225
226
  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.
227
- Use [torch.library.opcheck()](https://pytorch.org/docs/stable/library.html#torch.library.opcheck)
228
  to test the function registration and meta-function for any registered ops.
229
  See `tests/kernels` for examples.
230
231
232
- 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:
233
  [Custom Class Support in PT2](https://docs.google.com/document/d/18fBMPuOJ0fY5ZQ6YyrHUppw9FA332CpNtgB6SOIgyuA).
234

235
### Notes for Large Changes
236
237
238
239

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
240
it with `rfc-required` and might not go through the PR.
241

242
### What to Expect for the Reviews
243
244
245
246
247
248
249
250
251
252
253
254

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.
255
- After the review, the reviewer will put an `action-required` label on the PR
256
257
258
259
260
  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.
261
262
263
- 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.
264

265
## Thank You
266
267
268

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!