README.md 5.55 KB
Newer Older
1
# Kubernetes utilities for Dynamo Benchmarking and Profiling
2

3
4
5
6
7
8
This directory contains utilities and manifests for Dynamo benchmarking and profiling workflows.

## Prerequisites

**Before using these utilities, you must first set up Dynamo Cloud following the main installation guide:**

9
👉 **[Follow the Dynamo Cloud installation guide](/docs/kubernetes/installation_guide.md) to install the Dynamo Kubernetes Platform first.**
10
11
12
13
14

This includes:
1. Installing the Dynamo CRDs
2. Installing the Dynamo Platform (operator, etcd, NATS)
3. Setting up your target namespace
15
16
17

## Contents

18
- `setup_benchmarking_resources.sh` — Sets up benchmarking and profiling resources in your existing Dynamo namespace
19
- `manifests/`
20
  - `pvc.yaml` — PVC `dynamo-pvc` for storing profiler results and configurations
21
  - `pvc-access-pod.yaml` — short‑lived pod for copying profiler results from the PVC
22
23
24
25
26
- `kubernetes.py` — helper used by tooling to apply/read resources (e.g., access pod for PVC downloads)
- `inject_manifest.py` — utility for injecting deployment configurations into the PVC for profiling
- `download_pvc_results.py` — utility for downloading benchmark/profiling results from the PVC
- `dynamo_deployment.py` — utilities for working with DynamoGraphDeployment resources
- `requirements.txt` — Python dependencies for benchmarking utilities
27
28
29

## Quick start

30
### Benchmarking Resource Setup
31

32
After setting up Dynamo Cloud, use this script to prepare your namespace with the additional resources needed for benchmarking and profiling workflows:
33

34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
The setup script creates a `dynamo-pvc` with `ReadWriteMany` (RWX). If your cluster's default `storageClassName` does not support RWX, set `storageClassName` in `deploy/utils/manifests/pvc.yaml` to an RWX-capable class before running the script.

Example (add under `spec` in `deploy/utils/manifests/pvc.yaml`):
```yaml
...
spec:
  accessModes:
  - ReadWriteMany
  storageClassName: <your-rwx-storageclass>
...
```

> [!TIP]
> **Check your clusters storage classes**
>
> - List storage classes and provisioners:
> ```bash
> kubectl get sc -o wide
> ```

54
```bash
55
56
export NAMESPACE=your-dynamo-namespace
export HF_TOKEN=<HF_TOKEN>  # Optional: for HuggingFace model access
57

58
deploy/utils/setup_benchmarking_resources.sh
59
60
```

61
This script applies the following manifests to your existing Dynamo namespace:
62
63
64

- `deploy/utils/manifests/pvc.yaml` - PVC `dynamo-pvc`

65
If `HF_TOKEN` is provided, it also creates a secret for HuggingFace model access.
66

67
After running the setup script, verify the resources by checking:
68
69

```bash
70
kubectl get pvc dynamo-pvc -n $NAMESPACE
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
```

### PVC Manipulation Scripts

These scripts interact with the Persistent Volume Claim (PVC) that stores configuration files and benchmark/profiling results. They're essential for the Dynamo benchmarking and profiling workflows.

#### Why These Scripts Are Needed

1. **For Pre-Deployment Profiling**: The profiling job needs access to your Dynamo deployment configurations (DGD manifests) to test different parallelization strategies
2. **For Retrieving Results**: Both benchmarking and profiling jobs write their results to the PVC, which you need to download for analysis

#### Script Usage

**Inject deployment configurations for profiling:**

```bash
# The profiling job reads your DGD config from the PVC
88
89
# IMPORTANT: All paths must start with /data/ for security reasons
python3 -m deploy.utils.inject_manifest \
90
91
  --namespace $NAMESPACE \
  --src ./my-disagg.yaml \
92
  --dest /data/configs/disagg.yaml
93
94
```

95
**Download benchmark results:**
96
97

```bash
98
# After benchmarking completes, download results
99
python3 -m deploy.utils.download_pvc_results \
100
  --namespace $NAMESPACE \
101
  --output-dir ./benchmarks/results \
102
  --folder /data/results \
103
104
105
  --no-config   # optional: skip *.yaml/*.yml in the download
```

106
107
108
109
110
111
112
113
114
115
116
117
118
**Download profiling results (optional, for local inspection):**

```bash
# Optional: Download profiling data for local analysis
# The planner reads directly from the PVC, so this is only needed for inspection
python3 -m deploy.utils.download_pvc_results \
  --namespace $NAMESPACE \
  --output-dir ./profiling_data \
  --folder /data
```

> **Note on Profiling Results**: When using DGDR (DynamoGraphDeploymentRequest) for SLA-driven profiling, profiling data is stored in `/data/` on the PVC. The planner component reads this data directly from the PVC, so downloading is **optional** - only needed if you want to inspect the profiling results locally (e.g., view performance plots, check configurations).

119
120
121
122
123
124
#### Path Requirements

**Important**: The PVC is mounted at `/data` in the access pod for security reasons. All destination paths must start with `/data/`.

**Common path patterns:**
- `/data/configs/` - Configuration files (DGD manifests)
125
126
- `/data/results/` - Benchmark results (for download after benchmarking jobs)
- `/data/` - Profiling data (used directly by planner, typically not downloaded)
127
128
129
130
- `/data/benchmarking/` - Benchmarking artifacts

**User-friendly error messages**: If you forget the `/data/` prefix, the script will show a helpful error message with the correct path and example commands.

131
132
#### Next Steps

133
For complete benchmarking and profiling workflows:
134
- **Benchmarking Guide**: See [docs/benchmarks/benchmarking.md](../../docs/benchmarks/benchmarking.md) for comparing DynamoGraphDeployments and external endpoints
135
- **Pre-Deployment Profiling**: See [docs/benchmarks/sla_driven_profiling.md](../../docs/benchmarks/sla_driven_profiling.md) for optimizing configurations before deployment
136
137
138

## Notes

139
- This setup is focused on benchmarking and profiling resources only - the main Dynamo platform must be installed separately.