-
Notifications
You must be signed in to change notification settings - Fork 387
build: remove dependency install for nixl #594
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Dependencies are already present as part of cuda-dl-base image
f4ecb71
to
57962dd
Compare
Signed-off-by: Tim Stamler <tstamler@nvidia.com>
Signed-off-by: Tim Stamler <tstamler@nvidia.com>
are we concentrating on this request or this one: |
this one. closed 407 |
@@ -182,6 +73,9 @@ RUN mkdir /opt/dynamo && \ | |||
ENV VIRTUAL_ENV=/opt/dynamo/venv | |||
ENV PATH="${VIRTUAL_ENV}/bin:${PATH}" | |||
|
|||
# Install NIXL Python module | |||
RUN cd /opt/nixl && uv pip install . |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
For ARM support it looks like we need to set gds_path
explicitly because it's hard-coded on nixl side atm. This worked for me on a quick test until we can move the arch detection to NIXL side:
# Install NIXL Python module
# TODO: Can NIXL build select gds_path based on arch instead of us?
RUN if [ "$ARCH" = "arm64" ]; then \
cd /opt/nixl && uv pip install . --config-settings=setup-args="-Dgds_path=/usr/local/cuda/targets/sbsa-linux/"; \
else \
cd /opt/nixl && uv pip install .; \
fi
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This PR looks great to me otherwise!
I've included and extended this PR with ARM support in this PR: #839 |
Signed-off-by: Anant Sharma <anants@nvidia.com>
Overview:
Dependencies are already present as part of cuda-dl-base image
Details:
Where should the reviewer start?
Related Issues: (use one of the action keywords Closes / Fixes / Resolves / Relates to)