Skip to content

GitLab

Unverified Commit a3e6468d authored by Alec's avatar Alec Committed by GitHub
Browse files

feat(kvbm): build kvbm wheel with nccl feature for CUDA containers (#8120) 


Signed-off-by: default avatarAlec Flowers <aflowers@nvidia.com>
parent 36e8b8c8
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
container/templates/wheel_builder.Dockerfile
View file @ a3e6468d
...@@ -584,7 +584,9 @@ RUN --mount=type=secret,id=aws-key-id,env=AWS_ACCESS_KEY_ID \ ...@@ -584,7 +584,9 @@ RUN --mount=type=secret,id=aws-key-id,env=AWS_ACCESS_KEY_ID \
source ${VIRTUAL_ENV}/bin/activate && \ source ${VIRTUAL_ENV}/bin/activate && \
if [ "$ENABLE_KVBM" = "true" ]; then \ if [ "$ENABLE_KVBM" = "true" ]; then \
cd /opt/dynamo/lib/bindings/kvbm && \ cd /opt/dynamo/lib/bindings/kvbm && \
maturin build --release --out target/wheels && \ KVBM_FEATURES=""; \
if [ "$DEVICE" = "cuda" ]; then KVBM_FEATURES="--features nccl"; fi && \
maturin build --release ${KVBM_FEATURES} --out target/wheels && \
if [ "$DEVICE" = "cuda" ]; then \ if [ "$DEVICE" = "cuda" ]; then \
auditwheel repair \ auditwheel repair \
--exclude libnixl.so \ --exclude libnixl.so \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment