Skip to content

Commit d248695

Browse files
committed
ci(wheel): build with mm-rdma so the vLLM RDMA e2e exercises the real NIXL lane
The e2e wheel was built --features vendored-openssl only, so smg-mm-rdma compiled its no-op stub and every multimodal export fell back to inline — smg_mm_tensor_bytes_total{path="remote"} stayed 0 and test_single_image_uses_rdma failed. Forward an mm-rdma feature from smg-python to smg/mm-rdma and enable it in the wheel build so the gateway ships the real NIXL exporter. Signed-off-by: Simo Lin <linsimo.mark@gmail.com>
1 parent ec1fe40 commit d248695

2 files changed

Lines changed: 9 additions & 1 deletion

File tree

bindings/python/Cargo.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ workspace = true
3636
default = ["pyo3/extension-module"]
3737
opencv-video = ["smg/opencv-video"]
3838
vendored-openssl = ["smg/vendored-openssl"]
39+
mm-rdma = ["smg/mm-rdma"]
3940

4041
[profile.ci]
4142
inherits = "release"

scripts/ci_build_wheel.sh

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,13 @@ fi
1818
# Install maturin and zig for manylinux cross-compilation
1919
python3 -m pip install --upgrade pip maturin ziglang
2020

21+
# The mm-rdma feature pulls nixl-sys, which runs bindgen at build time and needs
22+
# libclang. Install it (Debian/Ubuntu runners) so binding generation can find it.
23+
if command -v apt-get >/dev/null 2>&1; then
24+
sudo apt-get update
25+
sudo apt-get install -y --no-install-recommends clang libclang-dev
26+
fi
27+
2128
# Start Go FFI build in background
2229
echo "Starting Go FFI build in background..."
2330
(cd bindings/golang && make build && echo "Go FFI: OK" && ls -la target/release/libsmg_go.*) &
@@ -26,7 +33,7 @@ GO_PID=$!
2633
# Build Python wheel in foreground
2734
echo "Building Python wheel..."
2835
cd bindings/python
29-
maturin build --profile ci --features vendored-openssl --manylinux 2_28 --zig --out dist
36+
maturin build --profile ci --features vendored-openssl,mm-rdma --manylinux 2_28 --zig --out dist
3037
echo "Python wheel: OK"
3138
ls -lh dist/
3239

0 commit comments

Comments
 (0)