vllm-project/vllm

[Feature]: Run performance benchmarks for multi-modal models in CI

オープン

#16,353 opened on 2025/04/09

 (15 件のコメント) (0 件のリアクション) (0 人の担当者)Python (16,816 件のフォーク)batch import
feature requesthelp wantedkeep-openmulti-modality

Repository metrics

Stars
 (80,034 個のスター)
PR merge metrics
 (平均マージ 3d 17h) (30d で 993 merged PRs)

説明

🚀 The feature, motivation and pitch

We currently only have benchmarks for text-only models such as Llama. With the increasing importance of multi-modality and related optimizations such as processor cache, we should add performance benchmarks for multi-modal models to avoid regressions (e.g. memory leaks, slow batching).

We can measure the peak memory usage based on this code:

import resource

max_self_usage = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / (1 << 20)
max_children_usage = resource.getrusage(resource.RUSAGE_CHILDREN).ru_maxrss / (1 << 20)
print(f"Peak memory usage: {max_self_usage} (self) + {max_children_usage} (children) GiB")

Alternatives

No response

Additional context

cc @mgoin @ywang96

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.

コントリビューターガイド