fix: update nixl arm build for other frameworks

Apply changes from #1544 to other dockfiles

Author: nv-anants

container/Dockerfile.sglang

@@ -85,21 +85,12 @@ WORKDIR /workspace
 # Copy nixl source, and use commit hash as cache hint
 COPY --from=nixl_base /opt/nixl /opt/nixl
 COPY --from=nixl_base /opt/nixl/commit.txt /opt/nixl/commit.txt
-RUN if [ "$ARCH" = "arm64" ]; then \
-        cd /opt/nixl && \
-        mkdir build && \
-        meson setup build/ --prefix=/usr/local/nixl -Dgds_path=/usr/local/cuda/targets/sbsa-linux && \
-        cd build/ && \
-        ninja && \
-        ninja install; \
-    else \
-        cd /opt/nixl && \
+RUN cd /opt/nixl && \
         mkdir build && \
         meson setup build/ --prefix=/usr/local/nixl && \
         cd build/ && \
         ninja && \
-        ninja install; \
-    fi
+        ninja install
 
 ### NATS & ETCD SETUP ###
 # nats
@@ -126,12 +117,7 @@ ENV VIRTUAL_ENV=/opt/dynamo/venv
 ENV PATH="${VIRTUAL_ENV}/bin:${PATH}"
 
 # Install NIXL Python module
-# TODO: Move gds_path selection based on arch into NIXL build
-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
+RUN cd /opt/nixl && uv pip install .
 
 # Install sglang
 # Once either 0.4.6post6 or 0.4.7 is released, we can switch back to using the published version

container/Dockerfile.tensorrt_llm

@@ -102,21 +102,12 @@ SHELL ["/bin/bash", "-c"]
 # Copy nixl source, and use commit hash as cache hint
 COPY --from=nixl_base /opt/nixl /opt/nixl
 COPY --from=nixl_base /opt/nixl/commit.txt /opt/nixl/commit.txt
-RUN if [ "$ARCH" = "arm64" ]; then \
-        cd /opt/nixl && \
-        mkdir build && \
-        meson setup build/ --prefix=/usr/local/nixl -Dgds_path=/usr/local/cuda/targets/sbsa-linux && \
-        cd build/ && \
-        ninja && \
-        ninja install; \
-    else \
-        cd /opt/nixl && \
+RUN cd /opt/nixl && \
         mkdir build && \
         meson setup build/ --prefix=/usr/local/nixl && \
         cd build/ && \
         ninja && \
-        ninja install; \
-    fi
+        ninja install
 
 ENV NIXL_PREFIX=/usr/local/nixl
 

container/Dockerfile.vllm_v1

@@ -120,21 +120,12 @@ WORKDIR /workspace
 # Copy nixl source, and use commit hash as cache hint
 COPY --from=nixl_base /opt/nixl /opt/nixl
 COPY --from=nixl_base /opt/nixl/commit.txt /opt/nixl/commit.txt
-RUN if [ "$ARCH" = "arm64" ]; then \
-        cd /opt/nixl && \
-        mkdir build && \
-        meson setup build/ --buildtype=release --prefix=/usr/local/nixl -Dgds_path=/usr/local/cuda/targets/sbsa-linux && \
-        cd build/ && \
-        ninja && \
-        ninja install; \
-    else \
-        cd /opt/nixl && \
+RUN cd /opt/nixl && \
         mkdir build && \
         meson setup build/ --buildtype=release --prefix=/usr/local/nixl && \
         cd build/ && \
         ninja && \
-        ninja install; \
-    fi
+        ninja install
 
 ### NATS & ETCD SETUP ###
 # nats
@@ -161,13 +152,7 @@ ENV VIRTUAL_ENV=/opt/dynamo/venv
 ENV PATH="${VIRTUAL_ENV}/bin:${PATH}"
 
 # Install NIXL Python module
-# TODO: Move gds_path selection based on arch into NIXL build
-RUN if [ "$ARCH" = "arm64" ]; then \
-        cd /opt/nixl && uv build . --out-dir /workspace/wheels/nixl \
-        --config-settings=setup-args="-Dgds_path=/usr/local/cuda/targets/sbsa-linux"; \
-    else \
-        cd /opt/nixl && uv build . --out-dir /workspace/wheels/nixl; \
-    fi
+RUN cd /opt/nixl && uv build . --out-dir /workspace/wheels/nixl
 
 # Install the wheel
 # TODO: Move NIXL wheel install to the wheel_builder stage