README.md 3.69 KB
Newer Older
Ryan Olson's avatar
Ryan Olson committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
<!--
SPDX-FileCopyrightText: Copyright (c) 2024-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

https://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.
-->

18
# Dynamo Python Bindings
Ryan Olson's avatar
Ryan Olson committed
19

20
Python bindings for the Dynamo runtime system, enabling distributed computing capabilities for machine learning workloads.
Ryan Olson's avatar
Ryan Olson committed
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

## 🚀 Quick Start

1. Install `uv`: https://docs.astral.sh/uv/#getting-started
```
curl -LsSf https://astral.sh/uv/install.sh | sh
```

2. Install `protoc` protobuf compiler: https://grpc.io/docs/protoc-installation/.

For example on an Ubuntu/Debian system:
```
apt install protobuf-compiler
```

3. Setup a virtualenv
37

Ryan Olson's avatar
Ryan Olson committed
38
39
40
41
42
43
```
uv venv
source .venv/bin/activate
uv pip install maturin
```

Neelay Shah's avatar
Neelay Shah committed
44
4. Build and install dynamo wheel
Ryan Olson's avatar
Ryan Olson committed
45
46
47
48
```
maturin develop --uv
```

49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
5. Experimental: To allow using mistral.rs and llama.cpp via the bindings, build with feature flags:

```
maturin develop --features mistralrs,llamacpp --release
```

`--release` is optional. It builds slower but the resulting library is significantly faster.

See `examples/cli/cli.py` for usage.

They will both be built for CUDA by default. If you see a runtime error `CUDA_ERROR_STUB_LIBRARY` this is because
the stub `libcuda.so` is earlier on the library search path than the real libcuda. Try removing the `rpath` from the library:

```
patchelf --set-rpath '' _core.cpython-312-x86_64-linux-gnu.so
```

If you include the `llamacpp` feature flag, `libllama.so` and `libggml.so` (and family) will need to be available at runtime.


69
## Run Examples
Ryan Olson's avatar
Ryan Olson committed
70

71
### Prerequisite
Ryan Olson's avatar
Ryan Olson committed
72

73
See [README.md](../../../docs/runtime/README.md#prerequisites).
Ryan Olson's avatar
Ryan Olson committed
74

75
### Hello World Example
Ryan Olson's avatar
Ryan Olson committed
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

1. Start 3 separate shells, and activate the virtual environment in each
```
source .venv/bin/activate
```

2. In one shell (shell 1), run example server the instance-1
```
python3 ./examples/hello_world/server.py
```

3. (Optional) In another shell (shell 2), run example the server instance-2
```
python3 ./examples/hello_world/server.py
```

4. In the last shell (shell 3), run the example client:
```
python3 ./examples/hello_world/client.py
```

If you run the example client in rapid succession, and you started more than
one server instance above, you should see the requests from the client being
distributed across the server instances in each server's output. If only one
server instance is started, you should see the requests go to that server
each time.
102

103
## Performance
104

105
The performance impacts of synchronizing the Python and Rust async runtimes
106
107
108
109
110
111
is a critical consideration when optimizing the performance of a highly
concurrent and parallel distributed system.

The Python GIL is a global critical section and is ultimately the death of
parallelism. To compound that, when Rust async futures become ready,
accessing the GIL on those async event loop needs to be considered carefully.
112
Under high load, accessing the GIL or performing CPU intensive tasks on
113
114
115
116
117
118
119
on the event loop threads can starve out other async tasks for CPU resources.
However, performing a `tokio::task::spawn_blocking` is not without overheads
as well.

If bouncing many small message back-and-forth between the Python and Rust
event loops where Rust requires GIL access, this is pattern where moving the
code from Python to Rust will give you significant gains.