You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: ROCm_performance.md
+4-2Lines changed: 4 additions & 2 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -1,10 +1,12 @@
1
-
# Overview of the optional performance features uinque to https://github.com/ROCm/vllm
1
+
# Overview of the optional performance features unique to <https://github.com/ROCm/vllm>
2
2
3
3
## Triton attention
4
-
The default attention function on ROCm is using triton attention kernel. To fallback to the https://github.com/ROCm/flash-attention implementation set up the following environment symbol:
4
+
5
+
The default attention function on ROCm is using triton attention kernel. To fallback to the <https://github.com/ROCm/flash-attention> implementation set up the following environment symbol:
5
6
`VLLM_USE_TRITON_FLASH_ATTN=0`
6
7
7
8
## Tunable ops
9
+
8
10
Pytorch tunable ops are supported.
9
11
Define the following environment symbol: `PYTORCH_TUNABLEOP_ENABLED=1` in order to enable both the runtime tuning and the subsequent use of tuned results. To only use the tuned results without tuning any newly encountered shapes, set `PYTORCH_TUNABLEOP_TUNING=0`
Copy file name to clipboardExpand all lines: docs/dev-docker/README.md
+13-1Lines changed: 13 additions & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -101,7 +101,7 @@ Supermicro AS-8125GS-TNMR2 with 2x AMD EPYC 9575F Processors, 2.25 TiB RAM, 8x A
101
101
102
102
The vllm-dev docker image should work with any model supported by vLLM. When running with FP8, AMD has quantized models available for a variety of popular models, or you can quantize models yourself using Quark. If needed, the vLLM benchmark scripts will automatically download models and then store them in a Hugging Face cache directory for reuse in future tests. Alternatively, you can choose to download the model to the cache (or to another directory on the system) in advance.
103
103
104
-
Many HuggingFace models, including Llama-3.1, have gated access. You will need to set up an account at (https://huggingface.co), search for the model of interest, and request access if necessary. You will also need to create a token for accessing these models from vLLM: open your user profile (https://huggingface.co/settings/profile), select "Access Tokens", press "+ Create New Token", and create a new Read token.
104
+
Many HuggingFace models, including Llama-3.1, have gated access. You will need to set up an account at <https://huggingface.co>, search for the model of interest, and request access if necessary. You will also need to create a token for accessing these models from vLLM: open your user profile <https://huggingface.co/settings/profile>, select "Access Tokens", press "+ Create New Token", and create a new Read token.
vLLM provides a number of engine options which can be changed to improve performance. Refer to the [vLLM Engine Args](https://docs.vllm.ai/en/stable/usage/engine_args.html) documentation for the complete list of vLLM engine options.
208
208
209
209
Below is a list of a few of the key vLLM engine arguments for performance; these can be passed to the vLLM benchmark scripts:
210
+
210
211
-**--max-model-len** : Maximum context length supported by the model instance. Can be set to a lower value than model configuration value to improve performance and gpu memory utilization.
211
212
-**--max-num-batched-tokens** : The maximum prefill size, i.e., how many prompt tokens can be packed together in a single prefill. Set to a higher value to improve prefill performance at the cost of higher gpu memory utilization. 65536 works well for LLama models.
212
213
-**--max-num-seqs** : The maximum decode batch size (default 256). Using larger values will allow more prompts to be processed concurrently, resulting in increased throughput (possibly at the expense of higher latency). If the value is too large, there may not be enough GPU memory for the KV cache, resulting in requests getting preempted. The optimal value will depend on the GPU memory, model size, and maximum context length.
@@ -450,6 +451,7 @@ You should see some performance improvement about the e2e latency.
450
451
`rocm/vllm-dev:main` image has experimental [AITER](https://github.com/ROCm/aiter) support, and can yield siginficant performance increase for some model/input/output/batch size configurations. To enable the feature make sure the following environment is set: `VLLM_ROCM_USE_AITER=1`, the default value is `0`. When building your own image follow the [Docker build steps](#Docker-manifest) using the [aiter_integration_final](https://github.com/ROCm/vllm/tree/aiter_integration_final) branch.
451
452
452
453
Some use cases include:
454
+
453
455
- amd/Mixtral-8x7B-Instruct-v0.1-FP8-KV
454
456
- amd/Mixtral-8x22B-Instruct-v0.1-FP8-KV
455
457
@@ -511,45 +513,55 @@ Use AITER release candidate branch instead:
511
513
## Changelog
512
514
513
515
20250605_aiter:
516
+
514
517
- Updated to ROCm 6.4.1 and vLLM v0.9.0.1
515
518
- AITER MHA
516
519
- IBM 3d kernel for unified attention
517
520
- Full graph capture for split attention
518
521
519
522
20250521_aiter:
523
+
520
524
- AITER V1 engine performance improvement
521
525
522
526
20250513_aiter:
527
+
523
528
- Out of memory bug fix
524
529
- PyTorch fixes
525
530
- Tunable ops fixes
526
531
527
532
20250410_aiter:
533
+
528
534
- 2-stage MoE
529
535
- MLA from AITER
530
536
531
537
20250325_aiter:
538
+
532
539
- Improved DeepSeek-V3/R1 performance
533
540
- Initial Gemma-3 enablement
534
541
- Detokenizer disablement
535
542
- Torch.compile support
536
543
537
544
20250305_aiter:
545
+
538
546
- AITER improvements
539
547
- Support for FP8 skinny GEMM
540
548
541
549
20250207_aiter:
550
+
542
551
- More performant AITER
543
552
- Bug fixes
544
553
545
554
20250205_aiter:
555
+
546
556
-[AITER](https://github.com/ROCm/aiter) support
547
557
- Performance improvement for custom paged attention
548
558
- Reduced memory overhead bug fix
549
559
550
560
20250124:
561
+
551
562
- Fix accuracy issue with 405B FP8 Triton FA
552
563
- Fixed accuracy issue with TP8
553
564
554
565
20250117:
566
+
555
567
-[Experimental DeepSeek-V3 and DeepSeek-R1 support](#running-deepseek-v3-and-deepseek-r1)
0 commit comments