README.md 4.8 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
96
97
98
```

**Download benchmark/profiling results:**

```bash
# After benchmarking or profiling completes, download results
99
python3 -m deploy.utils.download_pvc_results \
100
101
  --namespace $NAMESPACE \
  --output-dir ./pvc_files \
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
#### 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)
- `/data/results/` - Benchmark results
- `/data/profiling_results/` - Profiling data
- `/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.

118
119
#### Next Steps

120
For complete benchmarking and profiling workflows:
121
- **Benchmarking Guide**: See [docs/benchmarks/benchmarking.md](../../docs/benchmarks/benchmarking.md) for comparing DynamoGraphDeployments and external endpoints
122
- **Pre-Deployment Profiling**: See [docs/benchmarks/sla_driven_profiling.md](../../docs/benchmarks/sla_driven_profiling.md) for optimizing configurations before deployment
123
124
125

## Notes

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