README.md 2.8 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
<!-- # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License. -->

# Benchmarks

18
This directory contains benchmarking scripts and tools for performance evaluation of Dynamo deployments. The benchmarking framework is a wrapper around aiperf that makes it easy to benchmark DynamoGraphDeployments or other deployments with exposed endpoints.
19
20
21

## Quick Start

22
23
### Benchmark a Dynamo Deployment
First, deploy your DynamoGraphDeployment using the [deployment documentation](../components/backends/), then:
24
25

```bash
26
# Port-forward your deployment to http://localhost:8000
27
kubectl port-forward -n <namespace> svc/<frontend-service-name> 8000:8000 > /dev/null 2>&1 &
28

29
# Run benchmark
30
python3 -m benchmarks.utils.benchmark \
31
32
    --benchmark-name my-benchmark \
    --endpoint-url http://localhost:8000 \
33
    --model "<your-model>"
34

35
36
# Generate plots
python3 -m benchmarks.utils.plot --data-dir ./benchmarks/results
37
38
39

# Or plot only specific benchmark experiments
python3 -m benchmarks.utils.plot --data-dir ./benchmarks/results --benchmark-name my-benchmark
40
41
42
43
```

## Features

44
Benchmark any HTTP endpoints! The benchmarking framework supports:
45
46

**Flexible Configuration:**
47
48
- User-defined benchmark names using `--benchmark-name` flag
- Support for single endpoint benchmarking with `--endpoint-url` flag
49
- Customizable concurrency levels (configurable via CONCURRENCIES env var), sequence lengths, and models
50
- Automated performance plot generation with custom benchmark names
51

52
**Supported Backends:**
53
54
- DynamoGraphDeployments with port-forwarded endpoints
- External HTTP endpoints (for comparison with non-Dynamo backends or platforms)
55
56
57

## Installation

58
This is already included as part of the Dynamo container images. To install locally or standalone:
59
60
61
62
63

```bash
pip install -e .
```

64
65
66
## Data Generation Tools

This directory also includes lightweight tools for:
67
- Analyzing prefix-structured data (`datagen analyze`)
68
69
- Synthesizing structured data customizable for testing purposes (`datagen synthesize`)

70
71
72
73
74
Detailed information is provided in the `prefix_data_generator` directory.

## Comprehensive Guide

For detailed documentation, configuration options, and advanced usage, see the [complete benchmarking guide](../docs/benchmarks/benchmarking.md).